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.
|