Upload infer_qwen2_vl.py with huggingface_hub

infer_qwen2_vl.py

@@ -26,6 +26,13 @@ processor = AutoProcessor.from_pretrained(model_path)
 
 import glob
 from PIL import Image
+import argparse
+
+parser = argparse.ArgumentParser(description="Process a dataset with specific index range.")
+parser.add_argument("--batch_size", type=int, default = 1,help="batch size")
+#parser.add_argument("--index", type=int, default = 0,help="index")
+args = parser.parse_args()
+
 
 folder = "/inspire/hdd/ws-ba572160-47f8-4ca1-984e-d6bcdeb95dbb/a100-maybe/albus/ICCV_2025/qvq/dataset"
 
@@ -35,7 +42,7 @@ for img_path in glob.glob(f"{folder}/*.jpe"):
     images.append(img)
 
 num_image = len(images)
-begin, end, batch_size= 0, num_image, 4
+begin, end, batch_size= 0, num_image, args.batch_size
 messages = [
     {
         "role": "system",