Upload infer_qwen2_vl.py with huggingface_hub
Browse files- infer_qwen2_vl.py +1 -1
infer_qwen2_vl.py
CHANGED
|
@@ -73,7 +73,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
|
|
| 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 |
)
|
|
|
|
| 73 |
for idx,i in enumerate(batch):
|
| 74 |
#img = batch[i]
|
| 75 |
print('gain image successfully !')
|
| 76 |
+
messages[1]["content"][0]["image"] = '/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/qvq/dataset/' + i
|
| 77 |
text = processor.apply_chat_template(
|
| 78 |
messages, tokenize=False, add_generation_prompt=True
|
| 79 |
)
|