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
|
| 15 |
-
from
|
| 16 |
-
from
|
| 17 |
-
from
|
| 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):
|