Verify architecture for loaded Spandrel models

2025-08-04 11:12:35 +00:00 · 2023-12-30 16:37:03 +02:00
parent c756133541
commit 4ad0c0c0a8
8 changed files with 22 additions and 5 deletions
--- a/modules/codeformer_model.py
+++ b/modules/codeformer_model.py
@@ -37,6 +37,7 @@ class FaceRestorerCodeFormer(face_restoration_utils.CommonFaceRestoration):
            return modelloader.load_spandrel_model(
                model_path,
                device=devices.device_codeformer,
+                expected_architecture='CodeFormer',
            ).model
        raise ValueError("No codeformer model found")