pulling esrgan denoise strength through to the generate API.

This commit is contained in:
tyler 2023-02-10 20:16:08 -06:00 committed by blessedcoolant
parent c00155f6a4
commit d3c850104b

View File

@ -987,7 +987,7 @@ class Generate:
if len(upscale) < 2:
upscale.append(0.75)
image = self.esrgan.process(
image, upscale[1], seed, int(upscale[0]), upscale_denoise_str=upscale_denoise_str)
image, upscale[1], seed, int(upscale[0]), denoise_str=upscale_denoise_str)
else:
print(">> ESRGAN is disabled. Image not upscaled.")
except Exception as e: