Ver código fonte

Pass sigma to apply_refiner

drhead 1 ano atrás
pai
commit
09d2e58811
1 arquivos alterados com 1 adições e 1 exclusões
  1. 1 1
      modules/sd_samplers_cfg_denoiser.py

+ 1 - 1
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']