IVGSZ commited on
Commit
76c25f2
1 Parent(s): f77c3cd

Update flash_vstream/eval_video/model_msvd_qa_featuresloader.py

Browse files
flash_vstream/eval_video/model_msvd_qa_featuresloader.py CHANGED
@@ -11,10 +11,10 @@ from tqdm import tqdm
11
  from torch.utils.data import Dataset, DataLoader
12
  from safetensors.torch import load_file
13
 
14
- from llama_vstream.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
15
- from llama_vstream.conversation import conv_templates, SeparatorStyle
16
- from llama_vstream.model.builder import load_pretrained_model
17
- from llama_vstream.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
18
 
19
 
20
  def split_list(lst, n):
 
11
  from torch.utils.data import Dataset, DataLoader
12
  from safetensors.torch import load_file
13
 
14
+ from flash_vstream.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
15
+ from flash_vstream.conversation import conv_templates, SeparatorStyle
16
+ from flash_vstream.model.builder import load_pretrained_model
17
+ from flash_vstream.mm_utils import tokenizer_image_token, get_model_name_from_path, KeywordsStoppingCriteria
18
 
19
 
20
  def split_list(lst, n):