mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-24 10:51:03 -08:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #15414 from DrBiggusDickus/dev2

Browse source 
Fix CodeFormer weight
This commit is contained in:
AUTOMATIC1111 2024-03-31 16:53:23 +03:00 • committed by GitHub
commit 23ef5027c6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
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):
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)