Skip to content

S
Stable Diffusion Webui
Commit ea831807 authored by DrBiggusDickus's avatar DrBiggusDickus
Browse files
Options

fix CodeFormer weight

parent f1a6c5fe
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/codeformer_model.py
View file @ ea831807
...@@ -50,7 +50,7 @@ class FaceRestorerCodeFormer(face_restoration_utils.CommonFaceRestoration): ...@@ -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)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment