Merge pull request #3874 from cobryan05/extra_tweak

Extras Tab - Option to upscale before face fix, caching improvements

1 files changed, 9 insertions, 0 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 2541970d..7ddfe07e 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1119,6 +1119,9 @@ def create_ui(wrap_gradio_gpu_call):
                     codeformer_visibility = gr.Slider(minimum=0.0, maximum=1.0, step=0.001, label="CodeFormer visibility", value=0, interactive=modules.codeformer_model.have_codeformer)

                     codeformer_weight = gr.Slider(minimum=0.0, maximum=1.0, step=0.001, label="CodeFormer weight (0 = maximum effect, 1 = minimum effect)", value=0, interactive=modules.codeformer_model.have_codeformer)

 

+                with gr.Group():

+                    upscale_before_face_fix = gr.Checkbox(label='Upscale Before Restoring Faces', value=False)

+

                 submit = gr.Button('Generate', elem_id="extras_generate", variant='primary')

 

             with gr.Column(variant='panel'):

@@ -1152,6 +1155,7 @@ def create_ui(wrap_gradio_gpu_call):
                 extras_upscaler_1,

                 extras_upscaler_2,

                 extras_upscaler_2_visibility,

+                upscale_before_face_fix,

             ],

             outputs=[

                 result_images,

@@ -1174,6 +1178,11 @@ def create_ui(wrap_gradio_gpu_call):
             outputs=[init_img_with_mask],

         )

 

+        extras_image.change(

+            fn=modules.extras.clear_cache,

+            inputs=[], outputs=[]

+        )

+

     with gr.Blocks(analytics_enabled=False) as pnginfo_interface:

         with gr.Row().style(equal_height=False):

             with gr.Column(variant='panel'):