Upload infer_qwen2_vl.py with huggingface_hub
Browse files- infer_qwen2_vl.py +2 -2
infer_qwen2_vl.py
CHANGED
|
@@ -71,9 +71,9 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 71 |
print(f"data index range : {batch_idx} ~ {min(batch_idx + batch_size, end)}")
|
| 72 |
image_inputs_batch, video_inputs_batch = [], []
|
| 73 |
for idx,i in enumerate(batch):
|
| 74 |
-
img = batch[i]
|
| 75 |
print('gain image successfully !')
|
| 76 |
-
messages[1]["content"][0]["image"] =
|
| 77 |
text = processor.apply_chat_template(
|
| 78 |
messages, tokenize=False, add_generation_prompt=True
|
| 79 |
)
|
|
|
|
| 71 |
print(f"data index range : {batch_idx} ~ {min(batch_idx + batch_size, end)}")
|
| 72 |
image_inputs_batch, video_inputs_batch = [], []
|
| 73 |
for idx,i in enumerate(batch):
|
| 74 |
+
#img = batch[i]
|
| 75 |
print('gain image successfully !')
|
| 76 |
+
messages[1]["content"][0]["image"] = i
|
| 77 |
text = processor.apply_chat_template(
|
| 78 |
messages, tokenize=False, add_generation_prompt=True
|
| 79 |
)
|