StarCycle commited on
Commit
ceb57ea
1 Parent(s): 3c7ca8a

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 = './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
 
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_data/'
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