diff options
author | Aarni Koskela <akx@iki.fi> | 2024-01-03 20:38:49 +0000 |
---|---|---|
committer | Aarni Koskela <akx@iki.fi> | 2024-01-03 20:38:49 +0000 |
commit | 3d31d5c27beb433fa37b30f135ec06a278a87630 (patch) | |
tree | dd6083adcf2601a0f676346e454a14b6539cf9d2 /extensions-builtin/SwinIR/scripts/swinir_model.py | |
parent | dfdc51246c678b585e1bdfdb7d2f202b0ca0e362 (diff) | |
download | stable-diffusion-webui-gfx803-3d31d5c27beb433fa37b30f135ec06a278a87630.tar.gz stable-diffusion-webui-gfx803-3d31d5c27beb433fa37b30f135ec06a278a87630.tar.bz2 stable-diffusion-webui-gfx803-3d31d5c27beb433fa37b30f135ec06a278a87630.zip |
SwinIR: pass model.scale
Diffstat (limited to 'extensions-builtin/SwinIR/scripts/swinir_model.py')
-rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index 6a8e21b0..16bf9b79 100644 --- 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() |