Browse Source

fix CodeFormer weight

DrBiggusDickus 1 year ago
parent
commit
ea83180761
1 changed files with 1 additions and 1 deletions
  1. 1 1
      modules/codeformer_model.py

+ 1 - 1
modules/codeformer_model.py

@@ -50,7 +50,7 @@ class FaceRestorerCodeFormer(face_restoration_utils.CommonFaceRestoration):
 
 
         def restore_face(cropped_face_t):
         def restore_face(cropped_face_t):
             assert self.net is not None
             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)
         return self.restore_with_helper(np_image, restore_face)