Skip to content

Commit 82f9898

Browse files
haichuan1221ywang96
authored andcommitted
Feature/add benchmark testing (vllm-project#5947)
Co-authored-by: Roger Wang <ywang@roblox.com>
1 parent 7f1f931 commit 82f9898

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

benchmarks/benchmark_serving.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -483,10 +483,10 @@ def main(args: argparse.Namespace):
483483

484484
elif args.dataset_name == "random":
485485
input_requests = sample_random_requests(
486-
input_len=args.input_len,
487-
output_len=args.output_len,
486+
input_len=args.random_input_len,
487+
output_len=args.random_output_len,
488488
num_prompts=args.num_prompts,
489-
range_ratio=args.range_ratio,
489+
range_ratio=args.random_range_ratio,
490490
tokenizer=tokenizer,
491491
)
492492

0 commit comments

Comments
 (0)