Spaces:
Running
on
Zero
Running
on
Zero
WeichenFan
commited on
Commit
·
d6ca52a
1
Parent(s):
9286cfe
Add application file
Browse files- sd3_pipeline.py +1 -0
sd3_pipeline.py
CHANGED
@@ -1093,6 +1093,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1093 |
|
1094 |
alpha = optimized_scale(positive_flat,negative_flat)
|
1095 |
alpha = alpha.view(batch_size, 1, 1, 1)
|
|
|
1096 |
|
1097 |
if (i <= zero_steps) and use_zero_init:
|
1098 |
noise_pred = noise_pred_text*0.
|
|
|
1093 |
|
1094 |
alpha = optimized_scale(positive_flat,negative_flat)
|
1095 |
alpha = alpha.view(batch_size, 1, 1, 1)
|
1096 |
+
alpha.to(positive_flat.dtype)
|
1097 |
|
1098 |
if (i <= zero_steps) and use_zero_init:
|
1099 |
noise_pred = noise_pred_text*0.
|