kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-01 16:51:11 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix arg name in _test_trtllm_serve_multimodal_benchmark.py

Browse Source
This commit is contained in:
Xiwen Yu 2025-09-02 01:34:09 -07:00
parent 62a78973a8
commit 90ce7866a8
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unittest/llmapi/apps/_test_trtllm_serve_multimodal_benchmark.py
View File

@ -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(),
})],