IVGSZ commited on
Commit
f77c3cd
1 Parent(s): 72a8072

Update flash_vstream/eval_video/model_msvd_qa.py

Browse files
flash_vstream/eval_video/model_msvd_qa.py CHANGED
@@ -8,11 +8,11 @@ import argparse
8
  from tqdm import tqdm
9
  from decord import VideoReader, cpu
10
 
11
- from llama_vstream.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
12
- from llama_vstream.conversation import conv_templates, SeparatorStyle
13
- from llama_vstream.model.builder import load_pretrained_model
14
- from llama_vstream.utils import disable_torch_init
15
- from llama_vstream.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
16
 
17
 
18
  def split_list(lst, n):
 
8
  from tqdm import tqdm
9
  from decord import VideoReader, cpu
10
 
11
+ from flash_vstream.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
12
+ from flash_vstream.conversation import conv_templates, SeparatorStyle
13
+ from flash_vstream.model.builder import load_pretrained_model
14
+ from flash_vstream.utils import disable_torch_init
15
+ from flash_vstream.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
16
 
17
 
18
  def split_list(lst, n):