From d3c850104bd6a369f9d5a5668691dd24d9e775a4 Mon Sep 17 00:00:00 2001 From: tyler Date: Fri, 10 Feb 2023 20:16:08 -0600 Subject: [PATCH] pulling esrgan denoise strength through to the generate API. --- ldm/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ldm/generate.py b/ldm/generate.py index c242826f03..32a6a929a8 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -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: