diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-12-30 22:09:51 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-12-30 22:22:58 +0000 |
| commit | 777af661a21821994993df3ef566b01df2bb61a0 (patch) | |
| tree | 75ab896cf4ad8d433212b87e8ebf3664d0d434aa /extensions-builtin/SwinIR | |
| parent | c0ca6348e8489651df861a101142805c213c66a0 (diff) | |
| download | stable-diffusion-webui-gfx803-777af661a21821994993df3ef566b01df2bb61a0.tar.gz stable-diffusion-webui-gfx803-777af661a21821994993df3ef566b01df2bb61a0.tar.bz2 stable-diffusion-webui-gfx803-777af661a21821994993df3ef566b01df2bb61a0.zip | |
Be more clear about Spandrel model nomenclature
Diffstat (limited to 'extensions-builtin/SwinIR')
| -rw-r--r-- | extensions-builtin/SwinIR/scripts/swinir_model.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/extensions-builtin/SwinIR/scripts/swinir_model.py b/extensions-builtin/SwinIR/scripts/swinir_model.py index aae159af..95c7ec64 100644 --- a/extensions-builtin/SwinIR/scripts/swinir_model.py +++ b/extensions-builtin/SwinIR/scripts/swinir_model.py @@ -71,7 +71,7 @@ class UpscalerSwinIR(Upscaler): else: filename = path - model = modelloader.load_spandrel_model( + model_descriptor = modelloader.load_spandrel_model( filename, device=self._get_device(), dtype=devices.dtype, @@ -79,10 +79,10 @@ class UpscalerSwinIR(Upscaler): ) if getattr(opts, 'SWIN_torch_compile', False): try: - model = torch.compile(model) + model_descriptor.model.compile() except Exception: logger.warning("Failed to compile SwinIR model, fallback to JIT", exc_info=True) - return model + return model_descriptor def _get_device(self): return devices.get_device_for('swinir') |