diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py index 44b84618ec0..0b353353be2 100644 --- a/modules/codeformer_model.py +++ b/modules/codeformer_model.py @@ -50,7 +50,7 @@ def restore(self, np_image, w: float | None = None): 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)