x54-729 commited on
Commit
e67ed29
1 Parent(s): 3e9c774

Update build_mlp.py

Browse files
Files changed (1) hide show
  1. build_mlp.py +2 -2
build_mlp.py CHANGED
@@ -6,8 +6,8 @@ import os
6
  from transformers import CLIPVisionModel, CLIPImageProcessor, CLIPVisionConfig
7
 
8
 
9
- def build_vision_tower(path):
10
- vision_tower = os.path.join(path, 'clip_l_560_pro7b')
11
  return CLIPVisionTower(vision_tower)
12
 
13
 
 
6
  from transformers import CLIPVisionModel, CLIPImageProcessor, CLIPVisionConfig
7
 
8
 
9
+ def build_vision_tower():
10
+ vision_tower = "internlm/internlm2-wqx-vl-clip"
11
  return CLIPVisionTower(vision_tower)
12
 
13