diff options
Diffstat (limited to 'modules/codeformer_model.py')
-rw-r--r-- | modules/codeformer_model.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py index ceda4bab..44b84618 100644 --- 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")
|