make Image CFG Scale only show if instrutpix2pix model is loaded

1 files changed, 9 insertions, 2 deletions

diff --git a/modules/ui.py b/modules/ui.py
index f2f7de8b..f5df1ffe 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -765,8 +765,9 @@ def create_ui():
 

                     elif category == "cfg":

                         with FormGroup():

-                            cfg_scale = gr.Slider(minimum=1.0, maximum=30.0, step=0.5, label='CFG Scale', value=7.0, elem_id="img2img_cfg_scale")

-                            image_cfg_scale = gr.Slider(minimum=0, maximum=3.0, step=0.05, label='Image CFG Scale (for instruct-pix2pix models only)', value=1.5, elem_id="img2img_image_cfg_scale")

+                            with FormRow():

+                                cfg_scale = gr.Slider(minimum=1.0, maximum=30.0, step=0.5, label='CFG Scale', value=7.0, elem_id="img2img_cfg_scale")

+                                image_cfg_scale = gr.Slider(minimum=0, maximum=3.0, step=0.05, label='Image CFG Scale', value=1.5, elem_id="img2img_image_cfg_scale", visible=shared.sd_model and shared.sd_model.cond_stage_key == "edit")

                             denoising_strength = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label='Denoising strength', value=0.75, elem_id="img2img_denoising_strength")

 

                     elif category == "seed":

@@ -1594,6 +1595,12 @@ def create_ui():
                 outputs=[component, text_settings],

             )

 

+        text_settings.change(

+            fn=lambda: gr.update(visible=shared.sd_model and shared.sd_model.cond_stage_key == "edit"),

+            inputs=[],

+            outputs=[image_cfg_scale],

+        )

+

         button_set_checkpoint = gr.Button('Change checkpoint', elem_id='change_checkpoint', visible=False)

         button_set_checkpoint.click(

             fn=lambda value, _: run_settings_single(value, key='sd_model_checkpoint'),