Image CFG Added (Full Implementation)

Uses separate denoiser for edit (instruct-pix2pix) models

No impact to txt2img or regular img2img

"Image CFG Scale" will only apply to instruct-pix2pix models and metadata will only be added if using such model

1 files changed, 3 insertions, 0 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 5e34fb07..f2f7de8b 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -766,6 +766,7 @@ 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")

                             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":

@@ -861,6 +862,7 @@ def create_ui():
                     batch_count,

                     batch_size,

                     cfg_scale,

+                    image_cfg_scale,

                     denoising_strength,

                     seed,

                     subseed, subseed_strength, seed_resize_from_h, seed_resize_from_w, seed_checkbox,

@@ -947,6 +949,7 @@ def create_ui():
                 (sampler_index, "Sampler"),

                 (restore_faces, "Face restoration"),

                 (cfg_scale, "CFG scale"),

+                (image_cfg_scale, "Image CFG scale"),

                 (seed, "Seed"),

                 (width, "Size-1"),

                 (height, "Size-2"),