Update finetune_lora.sh

This commit is contained in:
qianyu chen
2024-07-17 14:37:18 +08:00
committed by GitHub
parent 6c58d38e9e
commit 091236ad34

View File

@@ -37,7 +37,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--tune_vision true \
--tune_llm false \
--use_lora true \
--lora_target_modules "llm\..*layers\.\d+\.self_attn\.(q_proj|k_proj)" \
--lora_target_modules "llm\..*layers\.\d+\.self_attn\.(q_proj|k_proj|v_proj|o_proj)" \
--model_max_length 2048 \
--max_slice_nums 9 \
--max_steps 10000 \