diff --git a/examples/qwen_image/model_training/lora/Qwen-Image-Distill-Full.sh b/examples/qwen_image/model_training/lora/Qwen-Image-Distill-Full.sh index 983638d..cb487a2 100644 --- a/examples/qwen_image/model_training/lora/Qwen-Image-Distill-Full.sh +++ b/examples/qwen_image/model_training/lora/Qwen-Image-Distill-Full.sh @@ -11,5 +11,4 @@ accelerate launch examples/qwen_image/model_training/train.py \ --lora_base_model "dit" \ --lora_target_modules "to_q,to_k,to_v,add_q_proj,add_k_proj,add_v_proj,to_out.0,to_add_out,img_mlp.net.2,img_mod.1,txt_mlp.net.2,txt_mod.1" \ --lora_rank 32 \ - --align_to_opensource_format \ --use_gradient_checkpointing diff --git a/examples/qwen_image/model_training/lora/Qwen-Image-EliGen.sh b/examples/qwen_image/model_training/lora/Qwen-Image-EliGen.sh index 99dbb7f..af861e6 100644 --- a/examples/qwen_image/model_training/lora/Qwen-Image-EliGen.sh +++ b/examples/qwen_image/model_training/lora/Qwen-Image-EliGen.sh @@ -12,7 +12,6 @@ accelerate launch examples/qwen_image/model_training/train.py \ --lora_base_model "dit" \ --lora_target_modules "to_q,to_k,to_v,add_q_proj,add_k_proj,add_v_proj,to_out.0,to_add_out,img_mlp.net.2,img_mod.1,txt_mlp.net.2,txt_mod.1" \ --lora_rank 32 \ - --align_to_opensource_format \ --extra_inputs "eligen_entity_masks,eligen_entity_prompts" \ --use_gradient_checkpointing \ --find_unused_parameters diff --git a/examples/qwen_image/model_training/lora/Qwen-Image.sh b/examples/qwen_image/model_training/lora/Qwen-Image.sh index 15084c2..f1198a5 100644 --- a/examples/qwen_image/model_training/lora/Qwen-Image.sh +++ b/examples/qwen_image/model_training/lora/Qwen-Image.sh @@ -11,7 +11,6 @@ accelerate launch examples/qwen_image/model_training/train.py \ --lora_base_model "dit" \ --lora_target_modules "to_q,to_k,to_v,add_q_proj,add_k_proj,add_v_proj,to_out.0,to_add_out,img_mlp.net.2,img_mod.1,txt_mlp.net.2,txt_mod.1" \ --lora_rank 32 \ - --align_to_opensource_format \ --use_gradient_checkpointing \ --dataset_num_workers 8 \ --find_unused_parameters