Spaces:

nyanko7
/

sd-diffusers-webui

Running on T4

nyanko7 commited on Jun 21, 2023

Commit

3eced2a

•

1 Parent(s): 15e4f70

Update modules/model.py

Files changed (1) hide show

modules/model.py CHANGED Viewed

@@ -543,8 +543,8 @@ class StableDiffusionPipeline(DiffusionPipeline):
                 noise_pred_text - noise_pred_uncond
             )
-            if guidance_rescale > 0.0:
-                noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=guidance_rescale)
             return noise_pred
@@ -715,8 +715,8 @@ class StableDiffusionPipeline(DiffusionPipeline):
                 noise_pred_text - noise_pred_uncond
             )
-            if guidance_rescale > 0.0:
-                noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=guidance_rescale)
             return noise_pred

                 noise_pred_text - noise_pred_uncond
             )
+            if guidance_scale > 0.0:
+                noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=guidance_scale)
             return noise_pred
                 noise_pred_text - noise_pred_uncond
             )
+            if guidance_scale > 0.0:
+                noise_pred = rescale_noise_cfg(noise_pred, noise_pred_text, guidance_rescale=guidance_scale)
             return noise_pred