aboutsummaryrefslogtreecommitdiffstats
path: root/modules/realesrgan_model.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-12-30 22:38:43 +0000
committerGitHub <noreply@github.com>2023-12-30 22:38:43 +0000
commitce21840a042b9454a136372ab2971c1f21ec51e0 (patch)
treed4960972aa80b444137cd598ece5c63fc2a0d26e /modules/realesrgan_model.py
parentae124439c4b67755125bcf5d7ac6886501f4c11d (diff)
parent777af661a21821994993df3ef566b01df2bb61a0 (diff)
downloadstable-diffusion-webui-gfx803-ce21840a042b9454a136372ab2971c1f21ec51e0.tar.gz
stable-diffusion-webui-gfx803-ce21840a042b9454a136372ab2971c1f21ec51e0.tar.bz2
stable-diffusion-webui-gfx803-ce21840a042b9454a136372ab2971c1f21ec51e0.zip
Merge pull request #14477 from akx/spandrel-type-fix
Be more clear about Spandrel model nomenclature and types
Diffstat (limited to 'modules/realesrgan_model.py')
-rw-r--r--modules/realesrgan_model.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/realesrgan_model.py b/modules/realesrgan_model.py
index 65f2e880..4d35b695 100644
--- a/modules/realesrgan_model.py
+++ b/modules/realesrgan_model.py
@@ -36,14 +36,14 @@ class UpscalerRealESRGAN(Upscaler):
errors.report(f"Unable to load RealESRGAN model {path}", exc_info=True)
return img
- mod = modelloader.load_spandrel_model(
+ model_descriptor = modelloader.load_spandrel_model(
info.local_data_path,
device=self.device,
half=(not cmd_opts.no_half and not cmd_opts.upcast_sampling),
expected_architecture="ESRGAN", # "RealESRGAN" isn't a specific thing for Spandrel
)
return upscale_with_model(
- mod,
+ model_descriptor,
img,
tile_size=opts.ESRGAN_tile,
tile_overlap=opts.ESRGAN_tile_overlap,