Commit 3d31d5c2 authored by Aarni Koskela's avatar Aarni Koskela

SwinIR: pass model.scale

parent dfdc5124
......@@ -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()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment