Upload infer_qwen2_vl.py with huggingface_hub
Browse files- infer_qwen2_vl.py +1 -1
infer_qwen2_vl.py
CHANGED
|
@@ -69,7 +69,7 @@ from tqdm import tqdm
|
|
| 69 |
ans = []
|
| 70 |
counter = 0
|
| 71 |
for batch_idx in tqdm(range(begin, end, batch_size)):
|
| 72 |
-
batch =
|
| 73 |
print(f"data index range : {batch_idx} ~ {min(batch_idx + batch_size, end)}")
|
| 74 |
image_inputs_batch, video_inputs_batch = [], []
|
| 75 |
for idx,i in enumerate(batch):
|
|
|
|
| 69 |
ans = []
|
| 70 |
counter = 0
|
| 71 |
for batch_idx in tqdm(range(begin, end, batch_size)):
|
| 72 |
+
batch = images[batch_idx: min(batch_idx + batch_size, end)]
|
| 73 |
print(f"data index range : {batch_idx} ~ {min(batch_idx + batch_size, end)}")
|
| 74 |
image_inputs_batch, video_inputs_batch = [], []
|
| 75 |
for idx,i in enumerate(batch):
|