Update llava_internlm2_chat_7b_dinov2_e1_gpu8_finetune.py
Browse files
llava_internlm2_chat_7b_dinov2_e1_gpu8_finetune.py
CHANGED
|
@@ -28,7 +28,7 @@ visual_encoder_name_or_path = 'facebook/dinov2-large'
|
|
| 28 |
pretrained_pth = './work_dirs/llava_internlm2_chat_7b_dinov2_e1_gpu8_pretrain_copy/epoch_1.pth' # noqa: E501
|
| 29 |
|
| 30 |
# Data
|
| 31 |
-
data_root = './'
|
| 32 |
data_path = data_root + 'LLaVA-Instruct-150K/llava_v1_5_mix665k.json'
|
| 33 |
image_folder = data_root + 'llava_images/'
|
| 34 |
prompt_template = PROMPT_TEMPLATE.internlm2_chat
|
|
|
|
| 28 |
pretrained_pth = './work_dirs/llava_internlm2_chat_7b_dinov2_e1_gpu8_pretrain_copy/epoch_1.pth' # noqa: E501
|
| 29 |
|
| 30 |
# Data
|
| 31 |
+
data_root = './data/LLaVA-Pretrain/'
|
| 32 |
data_path = data_root + 'LLaVA-Instruct-150K/llava_v1_5_mix665k.json'
|
| 33 |
image_folder = data_root + 'llava_images/'
|
| 34 |
prompt_template = PROMPT_TEMPLATE.internlm2_chat
|