Skip to content

Commit f071561

Browse files
authored
Merge pull request #301 from LLaVA-VL/yhzhang/llava_video_dev
chore: update checkpoint for llava-onevision-qwen2-72b-ov and llava-onevision-qwen2-7b-ov
2 parents 03b70e2 + acef4af commit f071561

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

scripts/video/train/SO400M_Qwen2_72B_ov_to_video_am9.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ echo "BASE_RUN_NAME: ${BASE_RUN_NAME}"
2626
# Stage 2
2727
PROMPT_VERSION="qwen_1_5"
2828
MID_RUN_NAME="llavanext-${VISION_MODEL_VERSION_CLEAN}-${LLM_VERSION_CLEAN}-ov_to_video_am9"
29-
PREV_STAGE_CHECKPOINT="lmms-lab/llava-onevision-qwen2-72b-ov"
29+
PREV_STAGE_CHECKPOINT="lmms-lab/llava-onevision-qwen2-72b-ov-si"
3030
echo "PREV_STAGE_CHECKPOINT: ${PREV_STAGE_CHECKPOINT}"
3131
echo "MID_RUN_NAME: ${MID_RUN_NAME}"
3232

scripts/video/train/SO400M_Qwen2_7B_ov_to_video_am9.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ echo "BASE_RUN_NAME: ${BASE_RUN_NAME}"
2626
# Stage 2
2727
PROMPT_VERSION="qwen_1_5"
2828
MID_RUN_NAME="llavanext-${VISION_MODEL_VERSION_CLEAN}-${LLM_VERSION_CLEAN}-ov_to_video_am9"
29-
PREV_STAGE_CHECKPOINT="lmms-lab/llava-onevision-qwen2-7b-ov"
29+
PREV_STAGE_CHECKPOINT="lmms-lab/llava-onevision-qwen2-7b-ov-si"
3030
echo "PREV_STAGE_CHECKPOINT: ${PREV_STAGE_CHECKPOINT}"
3131
echo "MID_RUN_NAME: ${MID_RUN_NAME}"
3232

@@ -75,7 +75,7 @@ ACCELERATE_CPU_AFFINITY=1 torchrun --nproc_per_node="${ARNOLD_WORKER_GPU}" --nno
7575
--torch_compile True \
7676
--torch_compile_backend "inductor" \
7777
--dataloader_drop_last True \
78-
--frames_upbound 110 \
78+
--frames_upbound 64 \
7979
--mm_newline_position grid \
8080
--add_time_instruction True \
8181
--force_sample True \

0 commit comments

Comments
 (0)