Spaces:
Runtime error
Runtime error
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)
|