Skip to content

Commit 4f2a2fe

Browse files
authored
Merge pull request #264 from LLaVA-VL/Luodian-patch-1
[script] add scripts for users to better understand training process.
2 parents 25f5b74 + b38f728 commit 4f2a2fe

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,4 +62,4 @@ ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${NUM_GPUS}" --nnodes="${NN
6262
--dataloader_drop_last True \
6363
--attn_implementation sdpa
6464

65-
# You can delete the sdpa attn_implementation if you want to use flash attn
65+
# You can delete the sdpa attn_implementation if you want to use flash attn

scripts/train/finetune_siglip_a4.sh renamed to scripts/train/direct_finetune_siglip_a4.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,4 +64,4 @@ ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${NUM_GPUS}" --nnodes="${NN
6464
--dataloader_drop_last True \
6565
--attn_implementation sdpa
6666

67-
# You can delete the sdpa attn_implementation if you want to use flash attn
67+
# You can delete the sdpa attn_implementation if you want to use flash attn

scripts/train/finetune_si.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${NUM_GPUS}" --nnodes="${NN
3030
--deepspeed scripts/zero3.json \
3131
--model_name_or_path $PREV_STAGE_CHECKPOINT \
3232
--version $PROMPT_VERSION \
33-
--data_path /mnt/bn/vl-research/workspace/boli01/projects/LLaVA_Next/scripts/i18n/scale_llms/next_ov_stage_july21.yaml \
33+
--data_path /mnt/bn/vl-research/workspace/boli01/projects/LLaVA_Next/scripts/i18n/scale_llms/next_3p2m_single_image.yaml \
3434
--image_folder /mnt/bn/vl-research/data/llava_data \
3535
--video_folder /mnt/bn/vl-research/data/llava_video \
3636
--mm_tunable_parts="mm_vision_tower,mm_mlp_adapter,mm_language_model" \

0 commit comments

Comments
 (0)