Change karras to kdiffusion

1 files changed, 16 insertions, 16 deletions

diff --git a/modules/ui.py b/modules/ui.py
index 28d4f1d1..fd5c0799 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -484,7 +484,7 @@ def create_ui():
                         with FormRow(elem_classes="checkboxes-row", variant="compact"):

                             restore_faces = gr.Checkbox(label='Restore faces', value=False, visible=len(shared.face_restorers) > 1, elem_id="txt2img_restore_faces")

                             tiling = gr.Checkbox(label='Tiling', value=False, elem_id="txt2img_tiling")

-                            t2i_enable_k_sched = gr.Checkbox(label='Custom Karras Scheduler', value=False, elem_id="txt2img_enable_k_sched")

+                            t2i_enable_k_sched = gr.Checkbox(label='Custom KDiffusion Scheduler', value=False, elem_id="txt2img_enable_k_sched")

                             enable_hr = gr.Checkbox(label='Hires. fix', value=False, elem_id="txt2img_enable_hr")

                             hr_final_resolution = FormHTML(value="", elem_id="txtimg_hr_finalres", label="Upscaled resolution", interactive=False)

 

@@ -511,9 +511,9 @@ def create_ui():
                                     with gr.Row():

                                         hr_negative_prompt = gr.Textbox(label="Negative prompt", elem_id="hires_neg_prompt", show_label=False, lines=3, placeholder="Negative prompt for hires fix pass.\nLeave empty to use the same negative prompt as in first pass.", elem_classes=["prompt"])

 

-                    elif category == "karras_scheduler":

-                        with FormGroup(visible=False, elem_id="txt2img_karras_scheduler") as t2i_k_sched_options:

-                            with FormRow(elem_id="txt2img_karras_scheduler_row1", variant="compact"):

+                    elif category == "kdiffusion_scheduler":

+                        with FormGroup(visible=False, elem_id="txt2img_kdiffusion_scheduler") as t2i_k_sched_options:

+                            with FormRow(elem_id="txt2img_kdiffusion_scheduler_row1", variant="compact"):

                                 t2i_k_sched_type = gr.Dropdown(label="Type", elem_id="t2i_k_sched_type", choices=['karras', 'exponential', 'polyexponential'], value='karras')

                                 t2i_k_sched_sigma_min = gr.Slider(minimum=0.0, maximum=0.5, step=0.05, label='sigma min', value=0.1, elem_id="txt2img_sigma_min")

                                 t2i_k_sched_sigma_max = gr.Slider(minimum=5.0, maximum=50.0, step=0.1, label='sigma max', value=10.0, elem_id="txt2img_sigma_max")

@@ -677,10 +677,10 @@ def create_ui():
                 (hr_negative_prompt, "Hires negative prompt"),

                 (hr_prompts_container, lambda d: gr.update(visible=True) if d.get("Hires prompt", "") != "" or d.get("Hires negative prompt", "") != "" else gr.update()),

                 (t2i_enable_k_sched, "Enable Custom Karras Schedule"),

-                (t2i_k_sched_type, "Karras Scheduler Type"),

-                (t2i_k_sched_sigma_max, "Karras Scheduler sigma_max"),

-                (t2i_k_sched_sigma_min, "Karras Scheduler sigma_min"),

-                (t2i_k_sched_rho, "Karras Scheduler rho"),

+                (t2i_k_sched_type, "KDiffusion Scheduler Type"),

+                (t2i_k_sched_sigma_max, "KDiffusion Scheduler sigma_max"),

+                (t2i_k_sched_sigma_min, "KDiffusion Scheduler sigma_min"),

+                (t2i_k_sched_rho, "KDiffusion Scheduler rho"),

                 *modules.scripts.scripts_txt2img.infotext_fields

             ]

             parameters_copypaste.add_paste_fields("txt2img", None, txt2img_paste_fields, override_settings)

@@ -872,11 +872,11 @@ def create_ui():
                         with FormRow(elem_classes="checkboxes-row", variant="compact"):

                             restore_faces = gr.Checkbox(label='Restore faces', value=False, visible=len(shared.face_restorers) > 1, elem_id="img2img_restore_faces")

                             tiling = gr.Checkbox(label='Tiling', value=False, elem_id="img2img_tiling")

-                            i2i_enable_k_sched = gr.Checkbox(label='Custom Karras Scheduler', value=False, elem_id="txt2img_enable_k_sched")

+                            i2i_enable_k_sched = gr.Checkbox(label='Custom KDiffusion Scheduler', value=False, elem_id="txt2img_enable_k_sched")

 

-                    elif category == "karras_scheduler":

-                        with FormGroup(visible=False, elem_id="img2img_karras_scheduler") as i2i_k_sched_options:

-                            with FormRow(elem_id="img2img_karras_scheduler_row1", variant="compact"):

+                    elif category == "kdiffusion_scheduler":

+                        with FormGroup(visible=False, elem_id="img2img_kdiffusion_scheduler") as i2i_k_sched_options:

+                            with FormRow(elem_id="img2img_kdiffusion_scheduler_row1", variant="compact"):

                                 i2i_k_sched_type = gr.Dropdown(label="Type", elem_id="t2i_k_sched_type", choices=['karras', 'exponential', 'polyexponential'], value='karras')

                                 i2i_k_sched_sigma_min = gr.Slider(minimum=0.0, maximum=0.5, step=0.05, label='sigma min', value=0.1, elem_id="txt2img_sigma_min")

                                 i2i_k_sched_sigma_max = gr.Slider(minimum=5.0, maximum=50.0, step=0.1, label='sigma max', value=10.0, elem_id="txt2img_sigma_max")

@@ -1091,10 +1091,10 @@ def create_ui():
                 (sampler_index, "Sampler"),

                 (restore_faces, "Face restoration"),

                 (i2i_enable_k_sched, "Enable Custom Karras Schedule"),

-                (i2i_k_sched_type, "Karras Scheduler Type"),

-                (i2i_k_sched_sigma_max, "Karras Scheduler sigma_max"),

-                (i2i_k_sched_sigma_min, "Karras Scheduler sigma_min"),

-                (i2i_k_sched_rho, "Karras Scheduler rho"),

+                (i2i_k_sched_type, "KDiffusion Scheduler Type"),

+                (i2i_k_sched_sigma_max, "KDiffusion Scheduler sigma_max"),

+                (i2i_k_sched_sigma_min, "KDiffusion Scheduler sigma_min"),

+                (i2i_k_sched_rho, "KDiffusion Scheduler rho"),

                 (cfg_scale, "CFG scale"),

                 (image_cfg_scale, "Image CFG scale"),

                 (seed, "Seed"),