diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 17:40:11 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 17:40:29 +0000 |
| commit | cefb5d6d7dbb35e68467bb7965f7139abfaf290d (patch) | |
| tree | 4392902a7ff807d67ae388f0fba3332e547a1b7c /extensions-builtin/SwinIR | |
| parent | 8fba733c0906dd3a80c0a3873793cffa4c78ce04 (diff) | |
| download | stable-diffusion-webui-gfx803-cefb5d6d7dbb35e68467bb7965f7139abfaf290d.tar.gz stable-diffusion-webui-gfx803-cefb5d6d7dbb35e68467bb7965f7139abfaf290d.tar.bz2 stable-diffusion-webui-gfx803-cefb5d6d7dbb35e68467bb7965f7139abfaf290d.zip | |
fix accessing options when they are not ready for SwinIR.
Diffstat (limited to 'extensions-builtin/SwinIR')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index 782769e2..9a74b253 100644 --- a/extensions-builtin/SwinIR/scripts/swinir_model.py +++ b/extensions-builtin/SwinIR/scripts/swinir_model.py @@ -99,11 +99,15 @@ class UpscalerSwinIR(Upscaler): def upscale( img, model, - tile=opts.SWIN_tile, - tile_overlap=opts.SWIN_tile_overlap, + tile=None, + tile_overlap=None, window_size=8, scale=4, ): + tile = tile or opts.SWIN_tile + tile_overlap = tile_overlap or opts.SWIN_tile_overlap + + img = np.array(img) img = img[:, :, ::-1] img = np.moveaxis(img, 2, 0) / 255 |