Pass sigma to apply_refiner

09d2e588 · drhead · GitHub · f4869f8d · 09d2e588
Commit 09d2e588 authored Feb 20, 2024 by drhead Committed by GitHub Feb 20, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

modules/sd_samplers_cfg_denoiser.py modules/sd_samplers_cfg_denoiser.py +1 -1

No files found.
--- 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']