Skip to content

S
Stable Diffusion Webui
Commit 3d31d5c2 authored by Aarni Koskela's avatar Aarni Koskela
Browse files
Options

SwinIR: pass model.scale

parent dfdc5124
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
extensions-builtin/SwinIR/scripts/swinir_model.py
View file @ 3d31d5c2
...@@ -51,7 +51,7 @@ class UpscalerSwinIR(Upscaler): ...@@ -51,7 +51,7 @@ class UpscalerSwinIR(Upscaler):
model, model,
tile_size=shared.opts.SWIN_tile, tile_size=shared.opts.SWIN_tile,
tile_overlap=shared.opts.SWIN_tile_overlap, tile_overlap=shared.opts.SWIN_tile_overlap,
scale=4, # TODO: This was hard-coded before too... scale=model.scale,
desc="SwinIR", desc="SwinIR",
) )
devices.torch_gc() 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