Quick fix for bug when no LoRA network weights is provided

This commit is contained in:
bmaltais 2023-01-01 14:37:32 -05:00
parent af46ce4c47
commit bfa590b313

View File

@ -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)