diff --git a/lora_gui.py b/lora_gui.py index bfef5229a..62319efe9 100644 --- a/lora_gui.py +++ b/lora_gui.py @@ -735,7 +735,7 @@ def train_model( ) return run_cmd += f' --network_module=lycoris.kohya' - run_cmd += f' --network_args "conv_dim={conv_dim}" "conv_alpha={conv_alpha}" "algo=lora"' + run_cmd += f' --network_args "conv_dim={conv_dim}" "conv_alpha={conv_alpha}" "algo=locon"' if LoRA_type == 'LyCORIS/LoHa': try: