From 90ce7866a881460d977459243745c76c45c2e510 Mon Sep 17 00:00:00 2001 From: Xiwen Yu Date: Tue, 2 Sep 2025 01:34:09 -0700 Subject: [PATCH] Fix arg name in _test_trtllm_serve_multimodal_benchmark.py --- .../apps/_test_trtllm_serve_multimodal_benchmark.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unittest/llmapi/apps/_test_trtllm_serve_multimodal_benchmark.py b/tests/unittest/llmapi/apps/_test_trtllm_serve_multimodal_benchmark.py index 502c366561..14146efa3d 100644 --- a/tests/unittest/llmapi/apps/_test_trtllm_serve_multimodal_benchmark.py +++ b/tests/unittest/llmapi/apps/_test_trtllm_serve_multimodal_benchmark.py @@ -72,11 +72,11 @@ def vision_arena_dataset_path(): @skip_gpu_memory_less_than_80gb @pytest.mark.parametrize("dataset_name,dataset_args", [("random_image", { - "--num-images": "1", - "--image-size": "512", + "--random-image-size": "1", + "--random-image-size": "512", }), ("random_image", { - "--num-images": "2", - "--image-size": "512", + "--random-num-images": "2", + "--random-image-size": "512", }), ("hf", { "--dataset-path": vision_arena_dataset_path(), })],