Merge pull request AUTOMATIC1111#7727 from missionfloyd/face-restore-…

…setting Fix face restorers setting
neutony · Feb 19, 2023 · 2a4f893 · 2a4f893
2 parents fb2354c + ceb8a4b
commit 2a4f893
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/modules/shared.py b/modules/shared.py
@@ -366,7 +366,7 @@ def list_samplers():
 }))
 
 options_templates.update(options_section(('face-restoration', "Face restoration"), {
-    "face_restoration_model": OptionInfo(None, "Face restoration model", gr.Radio, lambda: {"choices": [x.name() for x in face_restorers]}),
+    "face_restoration_model": OptionInfo("CodeFormer", "Face restoration model", gr.Radio, lambda: {"choices": [x.name() for x in face_restorers]}),
     "code_former_weight": OptionInfo(0.5, "CodeFormer weight parameter; 0 = maximum effect; 1 = minimum effect", gr.Slider, {"minimum": 0, "maximum": 1, "step": 0.01}),
     "face_restoration_unload": OptionInfo(False, "Move face restoration model from VRAM into RAM after processing"),
 }))

diff --git a/webui.py b/webui.py
@@ -98,7 +98,6 @@ def initialize():
     modules.sd_models.setup_model()
     codeformer.setup_model(cmd_opts.codeformer_models_path)
     gfpgan.setup_model(cmd_opts.gfpgan_models_path)
-    shared.face_restorers.append(modules.face_restoration.FaceRestoration())
 
     modelloader.list_builtin_upscalers()
     modules.scripts.load_scripts()