diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-08 12:19:36 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-08 12:19:36 +0000 |
| commit | f5001246e27e78422bb11187160702bcaba7daca (patch) | |
| tree | f61b564d088b5b2df74d86fae6c2ad4cb9cb7dd7 /modules/realesrgan_model.py | |
| parent | 3eea3c4dab96e75afcafbd37c3da5a31cac0f9cb (diff) | |
| download | stable-diffusion-webui-gfx803-f5001246e27e78422bb11187160702bcaba7daca.tar.gz stable-diffusion-webui-gfx803-f5001246e27e78422bb11187160702bcaba7daca.tar.bz2 stable-diffusion-webui-gfx803-f5001246e27e78422bb11187160702bcaba7daca.zip | |
honor tiling settings for RealESRGAN also
load scripts earlier to get errors before model loads
Diffstat (limited to 'modules/realesrgan_model.py')
| -rw-r--r-- | modules/realesrgan_model.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/realesrgan_model.py b/modules/realesrgan_model.py index e480887f..e2cef0c8 100644 --- a/modules/realesrgan_model.py +++ b/modules/realesrgan_model.py @@ -5,7 +5,7 @@ import numpy as np from PIL import Image
import modules.images
-from modules.shared import cmd_opts
+from modules.shared import cmd_opts, opts
RealesrganModelInfo = namedtuple("RealesrganModelInfo", ["name", "location", "model", "netscale"])
@@ -76,7 +76,9 @@ def upscale_with_realesrgan(image, RealESRGAN_upscaling, RealESRGAN_model_index) scale=info.netscale,
model_path=info.location,
model=model,
- half=not cmd_opts.no_half
+ half=not cmd_opts.no_half,
+ tile=opts.ESRGAN_tile,
+ tile_pad=opts.ESRGAN_tile_overlap,
)
upsampled = upsampler.enhance(np.array(image), outscale=RealESRGAN_upscaling)[0]
|