Skip to content

Commit 25f5b74

Browse files
authored
Update and rename finetune_onevision.sh to finetune_ov.sh
1 parent 1fcd3d4 commit 25f5b74

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed
Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@ echo "BASE_RUN_NAME: ${BASE_RUN_NAME}"
2020

2121
# Stage 2
2222
PROMPT_VERSION="qwen_1_5"
23-
MID_RUN_NAME="llava-onevision-${VISION_MODEL_VERSION_CLEAN}-${LLM_VERSION_CLEAN}-ov_stage_am9"
23+
RUN_NAME="llava-onevision-${VISION_MODEL_VERSION_CLEAN}-${LLM_VERSION_CLEAN}-ov_stage_am9"
2424
PREV_STAGE_CHECKPOINT="/mnt/bn/vl-research/checkpoints/onevision/llavanext-google_siglip-so400m-patch14-384-Qwen_Qwen2-7B-Instruct-mid_to_final_next_3m_am9_july14" # replace it with your last checkpoint training from single image collection
2525
echo "PREV_STAGE_CHECKPOINT: ${PREV_STAGE_CHECKPOINT}"
26-
echo "MID_RUN_NAME: ${MID_RUN_NAME}"
26+
echo "MID_RUN_NAME: ${RUN_NAME}"
2727

2828
ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${NUM_GPUS}" --nnodes="${NNODES}" --node_rank="${RANK}" --master_addr="${ADDR}" --master_port="${PORT}" \
2929
llava/train/train_mem.py \
@@ -45,8 +45,8 @@ ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${NUM_GPUS}" --nnodes="${NN
4545
--image_grid_pinpoints "(1x1),...,(6x6)" \
4646
--mm_patch_merge_type spatial_unpad \
4747
--bf16 True \
48-
--run_name $MID_RUN_NAME \
49-
--output_dir /mnt/bn/vl-research/checkpoints/onevision/$MID_RUN_NAME \
48+
--run_name $RUN_NAME \
49+
--output_dir /mnt/bn/vl-research/checkpoints/onevision/$RUN_NAME \
5050
--num_train_epochs 1 \
5151
--per_device_train_batch_size 1 \
5252
--per_device_eval_batch_size 4 \

0 commit comments

Comments
 (0)