Skip to content

Commit 5560900

Browse files
Isotr0pylk-chen
authored andcommitted
[Bugfix] Fix missing int type for -n in multi-image example (vllm-project#17223)
1 parent d7d3eea commit 5560900

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/offline_inference/vision_language_multi_image.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -791,7 +791,9 @@ def parse_args():
791791
parser.add_argument(
792792
"--num-images",
793793
"-n",
794-
choices=list(range(1, 13)), # 12 is the max number of images
794+
type=int,
795+
choices=list(range(1,
796+
len(IMAGE_URLS) + 1)), # the max number of images
795797
default=2,
796798
help="Number of images to use for the demo.")
797799
return parser.parse_args()

0 commit comments

Comments
 (0)