From 4f434c8aea3007cbf8038aec0c2a73c452ec9eca Mon Sep 17 00:00:00 2001 From: d8ahazard Date: Wed, 21 Sep 2022 08:38:38 -0500 Subject: [PATCH] Fix/Revert opts name from GAN to ESRGAN --- modules/esrgan_model.py | 4 ++-- modules/realesrgan_model.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/esrgan_model.py b/modules/esrgan_model.py index 930e790c6..7f3baf31f 100644 --- a/modules/esrgan_model.py +++ b/modules/esrgan_model.py @@ -92,10 +92,10 @@ def upscale_without_tiling(model, img): def esrgan_upscale(model, img): - if opts.GAN_tile == 0: + if opts.ESRGAN_tile == 0: return upscale_without_tiling(model, img) - grid = modules.images.split_grid(img, opts.GAN_tile, opts.GAN_tile, opts.GAN_tile_overlap) + grid = modules.images.split_grid(img, opts.ESRGAN_tile, opts.ESRGAN_tile, opts.ESRGAN_tile_overlap) newtiles = [] scale_factor = 1 diff --git a/modules/realesrgan_model.py b/modules/realesrgan_model.py index b24cffeb7..a39177735 100644 --- a/modules/realesrgan_model.py +++ b/modules/realesrgan_model.py @@ -96,8 +96,8 @@ def upscale_with_realesrgan(image, RealESRGAN_upscaling, RealESRGAN_model_index) model_path=info.location, model=model, half=not cmd_opts.no_half, - tile=opts.GAN_tile, - tile_pad=opts.GAN_tile_overlap, + tile=opts.ESRGAN_tile, + tile_pad=opts.ESRGAN_tile_overlap, ) upsampled = upsampler.enhance(np.array(image), outscale=RealESRGAN_upscaling)[0]