Merge pull request #23 from bmaltais/dev

Quick fix for bug when no LoRA network weights is provided
2023-01-01 14:38:07 -05:00 · 2023-01-01 14:38:07 -05:00 · 1a9554a800
commit 1a9554a800
parent 98f7dd5080 bfa590b313
1 changed files with 2 additions and 1 deletions
--- a/lora_gui.py
+++ b/lora_gui.py
@ -434,7 +434,8 @@ def train_model(
    # elif network_train == 'Unet only':
    #     run_cmd += f' --network_train_unet_only'
    run_cmd += f' --network_dim={network_dim}'
-    run_cmd += f' --network_weights={lora_network_weights}'
+    if not lora_network_weights == '':
+        run_cmd += f' --network_weights={lora_network_weights}'
    

    print(run_cmd)