Spaces:
Running
on
T4
Running
on
T4
Update modules/model.py
Browse files- modules/model.py +2 -2
modules/model.py
CHANGED
@@ -543,7 +543,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
|
|
543 |
noise_pred_text - noise_pred_uncond
|
544 |
)
|
545 |
|
546 |
-
noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=0.7)
|
547 |
return noise_pred
|
548 |
|
549 |
sampler_args = self.get_sampler_extra_args_i2i(sigma_sched, sampler)
|
@@ -713,7 +713,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
|
|
713 |
noise_pred_text - noise_pred_uncond
|
714 |
)
|
715 |
|
716 |
-
noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=0.7)
|
717 |
return noise_pred
|
718 |
|
719 |
extra_args = self.get_sampler_extra_args_t2i(
|
|
|
543 |
noise_pred_text - noise_pred_uncond
|
544 |
)
|
545 |
|
546 |
+
# noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=0.7)
|
547 |
return noise_pred
|
548 |
|
549 |
sampler_args = self.get_sampler_extra_args_i2i(sigma_sched, sampler)
|
|
|
713 |
noise_pred_text - noise_pred_uncond
|
714 |
)
|
715 |
|
716 |
+
# noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=0.7)
|
717 |
return noise_pred
|
718 |
|
719 |
extra_args = self.get_sampler_extra_args_t2i(
|