Update build_mlp.py
Browse files- 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(
|
10 |
-
vision_tower =
|
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 |
|