Verify architecture for loaded Spandrel models

2025-08-05 03:32:37 +00:00 · 2023-12-30 16:37:03 +02:00
parent c756133541
commit 4ad0c0c0a8
8 changed files with 22 additions and 5 deletions
--- a/extensions-builtin/SwinIR/scripts/swinir_model.py
+++ b/extensions-builtin/SwinIR/scripts/swinir_model.py
@@ -75,6 +75,7 @@ class UpscalerSwinIR(Upscaler):
            filename,
            device=self._get_device(),
            dtype=devices.dtype,
+            expected_architecture="SwinIR",
        )
        if getattr(opts, 'SWIN_torch_compile', False):
            try: