fix alphas cumprod (#478)

* fix alphas cumprod * indentation
lllyasviel · Mar 4, 2024 · b9705c5 · b9705c5
1 parent ce273ae
commit b9705c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/processing.py b/modules/processing.py
@@ -917,11 +917,13 @@ def rescale_zero_terminal_snr_abar(alphas_cumprod):
                 alphas_cumprod_backup = p.sd_model.alphas_cumprod
                 for modifier in alphas_cumprod_modifiers:
                     p.sd_model.alphas_cumprod = modifier(p.sd_model.alphas_cumprod)
+                p.sd_model.forge_objects.unet.model.model_sampling.set_sigmas(((1 - p.sd_model.alphas_cumprod) / p.sd_model.alphas_cumprod) ** 0.5)
 
             samples_ddim = p.sample(conditioning=p.c, unconditional_conditioning=p.uc, seeds=p.seeds, subseeds=p.subseeds, subseed_strength=p.subseed_strength, prompts=p.prompts)
 
             if alphas_cumprod_backup is not None:
                 p.sd_model.alphas_cumprod = alphas_cumprod_backup
+                p.sd_model.forge_objects.unet.model.model_sampling.set_sigmas(((1 - p.sd_model.alphas_cumprod) / p.sd_model.alphas_cumprod) ** 0.5)
 
             if p.scripts is not None:
                 ps = scripts.PostSampleArgs(samples_ddim)