Upload infer_qwen2_vl.py with huggingface_hub

infer_qwen2_vl.py

@@ -28,10 +28,10 @@ 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()
+# 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"
@@ -43,7 +43,7 @@ for img_path in glob.glob(f"{folder}/*.jpe"):
 
 num_image = len(images)
 print(f"beigin : {begin}, end : {end}, batch_size : {batch_size}")
-begin, end, batch_size= 0, 10, args.batch_size
+begin, end, batch_size= 0, 10, 2
 messages = [
     {
         "role": "system",