فهرست منبع

SwinIR: pass model.scale

Aarni Koskela 1 سال پیش
والد
کامیت
3d31d5c27b
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      extensions-builtin/SwinIR/scripts/swinir_model.py

+ 1 - 1
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()