Unverified Commit 2a4f8935 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #7727 from missionfloyd/face-restore-setting 

Fix face restorers setting
parents fb2354cb ceb8a4b2

modules/shared.py

+1 −1
Original line number Diff line number Diff line
@@ -366,7 +366,7 @@ options_templates.update(options_section(('upscaling', "Upscaling"), { 
}))



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"),

}))

webui.py

+0 −1
Original line number Diff line number Diff line
@@ -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()

CRA Git | Maintained and supported by SUSTech CRA and CCSE