Spaces:
Runtime error
Runtime error
LanguageBind
commited on
Commit
•
ebecb89
1
Parent(s):
1be81be
Update llava/model/multimodal_encoder/builder.py
Browse files
llava/model/multimodal_encoder/builder.py
CHANGED
@@ -22,7 +22,7 @@ def build_image_tower(image_tower_cfg, **kwargs):
|
|
22 |
|
23 |
def build_video_tower(video_tower_cfg, **kwargs):
|
24 |
video_tower = getattr(video_tower_cfg, 'mm_video_tower', getattr(video_tower_cfg, 'video_tower', None))
|
25 |
-
if video_tower.endswith('
|
26 |
return LanguageBindVideoTower(video_tower, args=video_tower_cfg, cache_dir='./cache_dir', **kwargs)
|
27 |
raise ValueError(f'Unknown video tower: {video_tower}')
|
28 |
|
|
|
22 |
|
23 |
def build_video_tower(video_tower_cfg, **kwargs):
|
24 |
video_tower = getattr(video_tower_cfg, 'mm_video_tower', getattr(video_tower_cfg, 'video_tower', None))
|
25 |
+
if video_tower.endswith('LanguageBind_Video_merge'):
|
26 |
return LanguageBindVideoTower(video_tower, args=video_tower_cfg, cache_dir='./cache_dir', **kwargs)
|
27 |
raise ValueError(f'Unknown video tower: {video_tower}')
|
28 |
|