Pass sigma to apply_refiner

2024-02-20 16:22:40 -05:00 · 2024-02-20 16:22:40 -05:00 · 09d2e58811
parent f4869f8de3
commit 09d2e58811
1 changed files with 1 additions and 1 deletions
--- a/modules/sd_samplers_cfg_denoiser.py
+++ b/modules/sd_samplers_cfg_denoiser.py
@ -152,7 +152,7 @@ class CFGDenoiser(torch.nn.Module):
        if state.interrupted or state.skipped:
            raise sd_samplers_common.InterruptedException

-        if sd_samplers_common.apply_refiner(self):
+        if sd_samplers_common.apply_refiner(self, sigma):
            cond = self.sampler.sampler_extra_args['cond']
            uncond = self.sampler.sampler_extra_args['uncond']