From f4667e564e1c53a835664650979cf08a4e9cace0 Mon Sep 17 00:00:00 2001 From: Lucas Freire Sangoi <125471877+DevArqSangoi@users.noreply.github.com> Date: Wed, 4 Oct 2023 02:24:26 -0300 Subject: [PATCH] Update lora_gui.py Changing the algo argument from "lora" to "locon" to match the option selected in the GUI for 'LoCon/LyCORIS'. The original repository specifies that when using kohya's training script, the algo should be set to "locon", not "lora". --- lora_gui.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: