fix missing arg

This commit is contained in:
drhead 2024-02-20 16:59:28 -05:00 committed by GitHub
parent 25eeeaa65f
commit bf348032bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -155,7 +155,7 @@ def replace_torchsde_browinan():
replace_torchsde_browinan()
def apply_refiner(cfg_denoiser):
def apply_refiner(cfg_denoiser, sigma):
if opts.refiner_switch_by_sample_steps:
completed_ratio = cfg_denoiser.step / cfg_denoiser.total_steps
else: