OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-05-15 16:29:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix CodeFormer weight

Browse Source
This commit is contained in:
DrBiggusDickus 2024-03-31 14:41:06 +02:00
parent f1a6c5fe17
commit ea83180761
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/codeformer_model.py
View File

@ -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)