Skip to content

Commit fbcc441

Browse files
czczupczczup
committed
Rename InternVL shell scripts
1 parent bfc8a3d commit fbcc441

File tree

7 files changed

+12
-12
lines changed

7 files changed

+12
-12
lines changed

‎internvl_chat_llava/README.md‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,18 +62,18 @@ Note: A better 336 resolution model will be released soon.
6262

6363
```
6464
# pretrain
65-
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/pretrain_internvit6b_resize_to_336_vicuna7b.sh
65+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/pretrain_internvit6b_224to336_vicuna7b.sh
6666
# finetune
67-
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/finetune_internvit6b_resize_to_336_vicuna7b.sh
67+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/finetune_internvit6b_224to336_vicuna7b.sh
6868
```
6969

7070
- InternViT-6B-224px + Vicuna-13B:
7171

7272
```
7373
# pretrain
74-
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/pretrain_internvit6b_resize_to_336_vicuna13b.sh
74+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/pretrain_internvit6b_224to336_vicuna13b.sh
7575
# finetune
76-
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/finetune_internvit6b_resize_to_336_vicuna13b.sh
76+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh scripts_internvl/finetune_internvit6b_224to336_vicuna13b.sh
7777
```
7878

7979
## 🤗 Model Zoo

‎internvl_chat_llava/llava/model/multimodal_encoder/clip_encoder.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,12 @@
1313

1414

1515
def is_intern_vit_6b_model(vision_tower_name):
16-
model_names = ["intern_vit_6b", "internvit_6b", "InternViT-6B"]
16+
model_names = ["intern_vit_6b", "internvit_6b", "InternViT-6B", "internvit6b"]
1717
return any(name in vision_tower_name for name in model_names)
1818

1919

2020
def is_internvl_14b_model(vision_tower_name):
21-
model_names = ["internvl_14b", "intern_vl_14b", "InternVL-14B"]
21+
model_names = ["internvl_14b", "intern_vl_14b", "InternVL-14B", "internvl14b"]
2222
return any(name in vision_tower_name for name in model_names)
2323

2424

internvl_chat_llava/scripts_internvl/finetune_internvit6b_resize_to_336_vicuna13b.sh renamed to internvl_chat_llava/scripts_internvl/finetune_internvit6b_224to336_vicuna13b.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
set -x
44

5-
OUTPUT_DIR='work_dirs/finetune_internvit6b_resize_to_336_vicuna13b'
5+
OUTPUT_DIR='work_dirs/finetune_internvit6b_224to336_vicuna13b'
66
GPUS=${GPUS:-8}
77
NNODES=${NNODES:-1}
88
NODE_RANK=${NODE_RANK:-0}

internvl_chat_llava/scripts_internvl/finetune_internvit6b_resize_to_336_vicuna13b_custom_data.sh renamed to internvl_chat_llava/scripts_internvl/finetune_internvit6b_224to336_vicuna13b_custom_data.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ QUOTA_TYPE="reserved"
99
NODES=$((GPUS / GPUS_PER_NODE))
1010
CPUS_PER_TASK=${CPUS_PER_TASK:-10}
1111
SRUN_ARGS=${SRUN_ARGS:-""}
12-
OUTPUT_DIR='work_dirs/finetune_internvit6b_resize_to_336_vicuna13b_custom_data'
12+
OUTPUT_DIR='work_dirs/finetune_internvit6b_224to336_vicuna13b_custom_data'
1313

1414
if [ ! -d "$OUTPUT_DIR" ]; then
1515
mkdir -p "$OUTPUT_DIR"
@@ -34,7 +34,7 @@ srun -p ${PARTITION} \
3434
--deepspeed ./scripts/zero1.json \
3535
--model_name_or_path ./pretrained/vicuna-13b-v1.5 \
3636
--version v1 \
37-
--meta_path ./scripts_internvl/meta/data_1108.json \
37+
--meta_path ./scripts_internvl/meta/custom_data.json \
3838
--vision_tower ./pretrained/intern_vit_6b_224px \
3939
--pretrain_mm_mlp_adapter ./work_dirs/pretrain_internvit6b_resize_to_336_vicuna13b/mm_projector.bin \
4040
--mm_projector_type mlp2x_gelu \

internvl_chat_llava/scripts_internvl/finetune_internvit6b_resize_to_336_vicuna7b.sh renamed to internvl_chat_llava/scripts_internvl/finetune_internvit6b_224to336_vicuna7b.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
set -x
44

5-
OUTPUT_DIR='work_dirs/finetune_internvit6b_resize_to_336_vicuna7b'
5+
OUTPUT_DIR='work_dirs/finetune_internvit6b_224to336_vicuna7b'
66
GPUS=${GPUS:-8}
77
NNODES=${NNODES:-1}
88
NODE_RANK=${NODE_RANK:-0}

internvl_chat_llava/scripts_internvl/pretrain_internvit6b_resize_to_336_vicuna13b.sh renamed to internvl_chat_llava/scripts_internvl/pretrain_internvit6b_224to336_vicuna13b.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
set -x
44

5-
OUTPUT_DIR='work_dirs/pretrain_internvit6b_resize_to_336_vicuna13b'
5+
OUTPUT_DIR='work_dirs/pretrain_internvit6b_224to336_vicuna13b'
66
GPUS=${GPUS:-8}
77
NNODES=${NNODES:-1}
88
NODE_RANK=${NODE_RANK:-0}

internvl_chat_llava/scripts_internvl/pretrain_internvit6b_resize_to_336_vicuna7b.sh renamed to internvl_chat_llava/scripts_internvl/pretrain_internvit6b_224to336_vicuna7b.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
set -x
44

5-
OUTPUT_DIR='work_dirs/pretrain_internvit6b_resize_to_336_vicuna7b'
5+
OUTPUT_DIR='work_dirs/pretrain_internvit6b_224to336_vicuna7b'
66
GPUS=${GPUS:-8}
77
NNODES=${NNODES:-1}
88
NODE_RANK=${NODE_RANK:-0}

0 commit comments

Comments
 (0)