Upload infer_qwen2_vl.py with huggingface_hub
Browse files- infer_qwen2_vl.py +2 -1
infer_qwen2_vl.py
CHANGED
@@ -78,12 +78,13 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
78 |
messages, tokenize=False, add_generation_prompt=True
|
79 |
)
|
80 |
image_inputs, video_inputs = process_vision_info(messages)
|
|
|
81 |
image_inputs_batch.append(image_inputs)
|
82 |
video_inputs_batch.append(video_inputs)
|
83 |
inputs = processor(
|
84 |
text=[text],
|
85 |
images=image_inputs_batch,
|
86 |
-
videos=
|
87 |
padding=True,
|
88 |
return_tensors="pt",
|
89 |
)
|
|
|
78 |
messages, tokenize=False, add_generation_prompt=True
|
79 |
)
|
80 |
image_inputs, video_inputs = process_vision_info(messages)
|
81 |
+
print(video_inputs)
|
82 |
image_inputs_batch.append(image_inputs)
|
83 |
video_inputs_batch.append(video_inputs)
|
84 |
inputs = processor(
|
85 |
text=[text],
|
86 |
images=image_inputs_batch,
|
87 |
+
videos=None,
|
88 |
padding=True,
|
89 |
return_tensors="pt",
|
90 |
)
|