From 3b93266aaec95315aa5bde24af3e154b5e33db2c Mon Sep 17 00:00:00 2001 From: bmaltais Date: Fri, 24 Feb 2023 07:30:37 -0500 Subject: [PATCH] Bug: Fix issue: https://github.com/bmaltais/kohya_ss/issues/231 --- library/merge_lora_gui.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/library/merge_lora_gui.py b/library/merge_lora_gui.py index d51fda2..12c2ea5 100644 --- a/library/merge_lora_gui.py +++ b/library/merge_lora_gui.py @@ -46,8 +46,8 @@ def merge_lora( run_cmd = f'.\\venv\Scripts\python.exe "networks\merge_lora.py"' run_cmd += f' --save_precision {save_precision}' run_cmd += f' --precision {precision}' - run_cmd += f' --save_to {save_to}' - run_cmd += f' --models {lora_a_model} {lora_b_model}' + run_cmd += f' --save_to "{save_to}"' + run_cmd += f' --models "{lora_a_model}" "{lora_b_model}"' run_cmd += f' --ratios {ratio_a} {ratio_b}' print(run_cmd) @@ -65,7 +65,7 @@ def gradio_merge_lora_tab(): with gr.Tab('Merge LoRA'): gr.Markdown('This utility can merge two LoRA networks together.') - lora_ext = gr.Textbox(value='*.pt *.safetensors', visible=False) + lora_ext = gr.Textbox(value='*.safetensors *.pt', visible=False) lora_ext_name = gr.Textbox(value='LoRA model types', visible=False) with gr.Row():