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('LanguageBind_Video'):
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