disable early cfg

models/pipeline.py

@@ -157,7 +157,7 @@ class SwittiPipeline:
             else:
                 freqs_cis = switti.freqs_cis
 
-            if si < turn_on_cfg_start_si or si >= turn_off_cfg_start_si:
+            if si >= turn_off_cfg_start_si:
                 apply_smooth = False
                 x_BLC = x_BLC[:B]
                 context = context[:B]
@@ -192,7 +192,8 @@ class SwittiPipeline:
 
             # Guidance
             if si < turn_on_cfg_start_si:
-                t = 0 # no guidance
+                # t = 0, i. e. no guidance
+                logits_BlV = logits_BlV[:B]
             elif si >= turn_on_cfg_start_si and si < turn_off_cfg_start_si:
                 # default const cfg
                 t = cfg