Upload infer_qwen2_vl.py with huggingface_hub
Browse files- infer_qwen2_vl.py +1 -1
infer_qwen2_vl.py
CHANGED
|
@@ -35,7 +35,7 @@ for img_path in glob.glob(f"{folder}/*.jpe"):
|
|
| 35 |
images.append(img)
|
| 36 |
|
| 37 |
num_image = len(images)
|
| 38 |
-
begin, end, batch_size= 0, num_image,
|
| 39 |
messages = [
|
| 40 |
{
|
| 41 |
"role": "system",
|
|
|
|
| 35 |
images.append(img)
|
| 36 |
|
| 37 |
num_image = len(images)
|
| 38 |
+
begin, end, batch_size= 0, num_image, 4
|
| 39 |
messages = [
|
| 40 |
{
|
| 41 |
"role": "system",
|