fix cfg to const

models/pipeline.py

@@ -164,10 +164,10 @@ class SwittiPipeline:
                 if crop_cond is not None:
                     crop_cond = crop_cond[:B]
                 for b in switti.blocks:
-                    if b.attn.caching:
+                    if b.attn.caching and b.attn.cached_k is not None:
                         b.attn.cached_k = b.attn.cached_k[:B]
                         b.attn.cached_v = b.attn.cached_v[:B]
-                    if b.cross_attn.caching:
+                    if b.cross_attn.caching  and b.cross_attn.cached_k is not None:
                         b.cross_attn.cached_k = b.cross_attn.cached_k[:B]
                         b.cross_attn.cached_v = b.cross_attn.cached_v[:B]
 
@@ -187,7 +187,7 @@ class SwittiPipeline:
 
             # Guidance
             if si < turn_off_cfg_start_si:
-                t = cfg * ratio
+                t = cfg
                 logits_BlV = (1 + t) * logits_BlV[:B] - t * logits_BlV[B:]
 
             if more_smooth and si >= smooth_start_si: