SwinIR: pass model.scale

tamu-edu · Jan 3, 2024 · 3d31d5c · 3d31d5c
1 parent dfdc512
commit 3d31d5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py
@@ -51,7 +51,7 @@ def do_upscale(self, img: Image.Image, model_file: str) -> Image.Image:
             model,
             tile_size=shared.opts.SWIN_tile,
             tile_overlap=shared.opts.SWIN_tile_overlap,
-            scale=4,  # TODO: This was hard-coded before too...
+            scale=model.scale,
             desc="SwinIR",
         )
         devices.torch_gc()