From ea83180761d450690cf590d4fd9b582241a9846d Mon Sep 17 00:00:00 2001
From: DrBiggusDickus <DrBiggusDickus>
Date: Sun, 31 Mar 2024 14:41:06 +0200
Subject: [PATCH] fix CodeFormer weight

---
 modules/codeformer_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)