Pass sigma to apply_refiner

This commit is contained in:
drhead
2024-02-20 16:22:40 -05:00
committed by GitHub
parent f4869f8de3
commit 09d2e58811

View File

@@ -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']