fix krea typo

This commit is contained in:
mi804
2025-08-01 17:13:45 +08:00
parent b63f81c6e3
commit b2554db100
10 changed files with 16 additions and 16 deletions

View File

@@ -7,7 +7,7 @@ accelerate launch examples/flux/model_training/train.py \
--learning_rate 1e-4 \
--num_epochs 5 \
--remove_prefix_in_ckpt "pipe.dit." \
--output_path "./models/train/FLUX.1-Kera-dev_lora" \
--output_path "./models/train/FLUX.1-Krea-dev_lora" \
--lora_base_model "dit" \
--lora_target_modules "a_to_qkv,b_to_qkv,ff_a.0,ff_a.2,ff_b.0,ff_b.2,a_to_out,b_to_out,proj_out,norm.linear,norm1_a.linear,norm1_b.linear,to_qkv_mlp" \
--lora_rank 32 \