Wendy-Fly commited on
Commit
4369fe7
·
verified ·
1 Parent(s): 36e4e7e

Upload infer_qwen2_vl.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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=video_inputs_batch,
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
  )