mazpie commited on
Commit
67ebba2
1 Parent(s): 8131f9b

Update third_party/InternVideo/InternVideo2/multi_modality/demo/small_utils.py

Browse files
third_party/InternVideo/InternVideo2/multi_modality/demo/small_utils.py CHANGED
@@ -81,7 +81,7 @@ def retrieve_text(frames,
81
 
82
  def setup_internvideo2(config: dict):
83
  if "bert" in config.model.text_encoder.name:
84
- tokenizer = BertTokenizer.from_pretrained(config.model.text_encoder.pretrained, local_files_only=True)
85
  model = InternVideo2_Stage2(config=config, tokenizer=tokenizer, is_pretrain=True)
86
  else:
87
  model = InternVideo2_Stage2(config=config, is_pretrain=True)
 
81
 
82
  def setup_internvideo2(config: dict):
83
  if "bert" in config.model.text_encoder.name:
84
+ tokenizer = BertTokenizer.from_pretrained(config.model.text_encoder.pretrained,)
85
  model = InternVideo2_Stage2(config=config, tokenizer=tokenizer, is_pretrain=True)
86
  else:
87
  model = InternVideo2_Stage2(config=config, is_pretrain=True)