SwinIR: pass model.scale

2024-01-03 22:38:49 +02:00 · 2024-01-03 22:38:49 +02:00 · 3d31d5c27b
parent dfdc51246c
commit 3d31d5c27b
1 changed files with 1 additions and 1 deletions
--- a/extensions-builtin/SwinIR/scripts/swinir_model.py
+++ b/extensions-builtin/SwinIR/scripts/swinir_model.py
@ -51,7 +51,7 @@ class UpscalerSwinIR(Upscaler):
            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()