Skip to content

Commit 1323d9d

Browse files
committed
raise error if min_num_images > max_num_images
Signed-off-by: breno.skuk <breno.skuk@hcompany.ai>
1 parent 5b8e941 commit 1323d9d

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

vllm/benchmarks/datasets.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -547,10 +547,11 @@ def get_image_sampling_params(
547547
limit_images_per_prompt)
548548
# ensure min num images is zero
549549
min_num_images = max(int(num_images * (1 - num_images_range_ratio)), 0)
550-
# assert min_num_images <= max_num_images
551-
assert min_num_images <= max_num_images, (
552-
"min_num_images must be <= max_num_images"
553-
)
550+
# raise error if min_num_images > max_num_images
551+
if min_num_images > max_num_images:
552+
raise ValueError(
553+
"min_num_images must be <= max_num_images"
554+
)
554555
# Enforce dimension_range_ratio < 1
555556
assert dimension_range_ratio < 1.0, (
556557
"dimension_range_ratio must be < 1.0 to ensure a valid sampling "

0 commit comments

Comments
 (0)