fix CodeFormer weight

2024-03-31 14:41:06 +02:00 · 2024-03-31 14:41:06 +02:00 · ea83180761
parent f1a6c5fe17
commit ea83180761
1 changed files with 1 additions and 1 deletions
--- a/modules/codeformer_model.py
+++ b/modules/codeformer_model.py
@ -50,7 +50,7 @@ class FaceRestorerCodeFormer(face_restoration_utils.CommonFaceRestoration):

        def restore_face(cropped_face_t):
            assert self.net is not None
-            return self.net(cropped_face_t, w=w, adain=True)[0]
+            return self.net(cropped_face_t, weight=w, adain=True)[0]

        return self.restore_with_helper(np_image, restore_face)