diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-04 08:17:20 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-04 08:17:20 +0000 |
| commit | 3f7f61e5411320a802a2a4b1afd38c8affee94b3 (patch) | |
| tree | 5473de8de4a7ffa71762b33b36d38f3ca92b821f /extensions-builtin/SwinIR/scripts | |
| parent | 1e7a8ce5e403de4bef7e09f522a48ce5c1b1d845 (diff) | |
| parent | 62470ee23443cb2ad3943a152ccae26a689c86e1 (diff) | |
| download | stable-diffusion-webui-gfx803-3f7f61e5411320a802a2a4b1afd38c8affee94b3.tar.gz stable-diffusion-webui-gfx803-3f7f61e5411320a802a2a4b1afd38c8affee94b3.tar.bz2 stable-diffusion-webui-gfx803-3f7f61e5411320a802a2a4b1afd38c8affee94b3.zip | |
Merge pull request #14524 from akx/fix-swinir-issues
Fix SwinIR issues
Diffstat (limited to 'extensions-builtin/SwinIR/scripts')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index bc427fea..16bf9b79 100644 --- a/extensions-builtin/SwinIR/scripts/swinir_model.py +++ b/extensions-builtin/SwinIR/scripts/swinir_model.py @@ -1,6 +1,7 @@ import logging import sys +import torch from PIL import Image from modules import devices, modelloader, script_callbacks, shared, upscaler_utils @@ -50,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() @@ -69,7 +70,7 @@ class UpscalerSwinIR(Upscaler): model_descriptor = modelloader.load_spandrel_model( filename, device=self._get_device(), - dtype=devices.dtype, + prefer_half=(devices.dtype == torch.float16), expected_architecture="SwinIR", ) if getattr(shared.opts, 'SWIN_torch_compile', False): |