Merge pull request #14473 from akx/soften-model-arch-check

Soften Spandrel model-architecture check to just a warning
AUTOMATIC1111 · Dec 30, 2023 · a86f441 · a86f441
2 parents 16848f9 + af050dc
commit a86f441
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/modules/modelloader.py b/modules/modelloader.py
@@ -196,7 +196,9 @@ def load_spandrel_model(
     import spandrel
     model = spandrel.ModelLoader(device=device).load_from_file(path)
     if expected_architecture and model.architecture != expected_architecture:
-        raise TypeError(f"Model {path} is not a {expected_architecture} model")
+        logger.warning(
+            f"Model {path!r} is not a {expected_architecture!r} model (got {model.architecture!r})",
+        )
     if half:
         model = model.model.half()
     if dtype: