Update llava_internlm2_chat_1_8b_clip_vit_large_p14_336_e1_gpu1_pretrain.py
Browse files
llava_internlm2_chat_1_8b_clip_vit_large_p14_336_e1_gpu1_pretrain.py
CHANGED
|
@@ -25,7 +25,7 @@ llm_name_or_path = 'internlm/internlm2-chat-1_8b'
|
|
| 25 |
visual_encoder_name_or_path = 'openai/clip-vit-large-patch14-336'
|
| 26 |
|
| 27 |
# Data
|
| 28 |
-
data_root = './llava_data/'
|
| 29 |
data_path = data_root + 'LLaVA-Pretrain/blip_laion_cc_sbu_558k.json'
|
| 30 |
image_folder = data_root + 'LLaVA-Pretrain/images'
|
| 31 |
prompt_template = PROMPT_TEMPLATE.internlm2_chat
|
|
|
|
| 25 |
visual_encoder_name_or_path = 'openai/clip-vit-large-patch14-336'
|
| 26 |
|
| 27 |
# Data
|
| 28 |
+
data_root = './data/llava_data/'
|
| 29 |
data_path = data_root + 'LLaVA-Pretrain/blip_laion_cc_sbu_558k.json'
|
| 30 |
image_folder = data_root + 'LLaVA-Pretrain/images'
|
| 31 |
prompt_template = PROMPT_TEMPLATE.internlm2_chat
|