(feat): Rework Checkpoint Merger UI for better clarity

- Rename variables to better align with field input
- Improve merged checkpoint output file name

1 files changed, 6 insertions, 5 deletions

diff --git a/modules/ui.py b/modules/ui.py
index e96109c9..e7382ca8 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -859,10 +859,11 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo, run_modelmerger):
             with gr.Column(variant='panel'):

                 gr.HTML(value="<p>A merger of the two checkpoints will be generated in your <b>/models</b> directory.</p>")

                 

-                modelname_0 = gr.Textbox(elem_id="modelmerger_modelname_0", label="Model Name (to)")

-                modelname_1 = gr.Textbox(elem_id="modelmerger_modelname_1", label="Model Name (from)")

-                interp_method = gr.Radio(choices=["Weighted Sum", "Sigmoid"], value="Weighted Sum", label="Interpolation Method")

+                with gr.Row():

+                    from_model_name = gr.Textbox(elem_id="modelmerger_from_model_name", label="Model Name (from)")

+                    to_model_name = gr.Textbox(elem_id="modelmerger_to_model_name", label="Model Name (to)")

                 interp_amount = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, label='Interpolation Amount', value=0.3)

+                interp_method = gr.Radio(choices=["Weighted Sum", "Sigmoid"], value="Weighted Sum", label="Interpolation Method")

                 submit = gr.Button(elem_id="modelmerger_merge", label="Merge", variant='primary')

             

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

@@ -871,8 +872,8 @@ def create_ui(txt2img, img2img, run_extras, run_pnginfo, run_modelmerger):
             submit.click(

                 fn=run_modelmerger,

                 inputs=[

-                    modelname_0,

-                    modelname_1,

+                    from_model_name,

+                    to_model_name,

                     interp_method,

                     interp_amount

                 ],