From 8c1c926d00623ea535bd402b8e11ce496621f2cc Mon Sep 17 00:00:00 2001 From: Isotr0py Date: Sat, 26 Apr 2025 16:49:52 +0800 Subject: [PATCH] [Bugfix] Fix missing int type for `-n` in multi-image example (#17223) --- examples/offline_inference/vision_language_multi_image.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/offline_inference/vision_language_multi_image.py b/examples/offline_inference/vision_language_multi_image.py index f165ea9efa10f..7f6608559f9c4 100644 --- a/examples/offline_inference/vision_language_multi_image.py +++ b/examples/offline_inference/vision_language_multi_image.py @@ -791,7 +791,9 @@ def parse_args(): parser.add_argument( "--num-images", "-n", - choices=list(range(1, 13)), # 12 is the max number of images + type=int, + choices=list(range(1, + len(IMAGE_URLS) + 1)), # the max number of images default=2, help="Number of images to use for the demo.") return parser.parse_args()