Spaces:
Sleeping
Sleeping
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,
|
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)
|