Spaces:

weepiess2383
/

CFG-Zero-Star

Running on Zero

WeichenFan commited on Mar 22

Commit

adf43eb

1 Parent(s): d6ca52a

Add application file

Files changed (1) hide show

sd3_pipeline.py CHANGED Viewed

@@ -1093,7 +1093,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
                         alpha = optimized_scale(positive_flat,negative_flat)
                         alpha = alpha.view(batch_size, 1, 1, 1)
-                        alpha.to(positive_flat.dtype)
                         if (i <= zero_steps) and use_zero_init:
                             noise_pred = noise_pred_text*0.

                         alpha = optimized_scale(positive_flat,negative_flat)
                         alpha = alpha.view(batch_size, 1, 1, 1)
+                        alpha = alpha.to(positive_flat.dtype)
                         if (i <= zero_steps) and use_zero_init:
                             noise_pred = noise_pred_text*0.