瀏覽代碼

Pass sigma to apply_refiner

drhead 1 年之前
父節點
當前提交
09d2e58811
共有 1 個文件被更改,包括 1 次插入1 次删除
  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:
         if state.interrupted or state.skipped:
             raise sd_samplers_common.InterruptedException
             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']
             cond = self.sampler.sampler_extra_args['cond']
             uncond = self.sampler.sampler_extra_args['uncond']
             uncond = self.sampler.sampler_extra_args['uncond']