Update build_mlp.py
Browse files- build_mlp.py +1 -2
build_mlp.py
CHANGED
|
@@ -7,8 +7,7 @@ from transformers import CLIPVisionModel
|
|
| 7 |
|
| 8 |
|
| 9 |
def build_vision_tower():
|
| 10 |
-
|
| 11 |
-
vision_tower = '/mnt/petrelfs/share_data/dongxiaoyi/share_models/clip_l_336'
|
| 12 |
return CLIPVisionTower(vision_tower)
|
| 13 |
|
| 14 |
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
def build_vision_tower():
|
| 10 |
+
vision_tower = 'openai/clip-vit-large-patch14-336'
|
|
|
|
| 11 |
return CLIPVisionTower(vision_tower)
|
| 12 |
|
| 13 |
|