Skip to content

Commit

Permalink
Feature/add benchmark testing (#5947)
Browse files Browse the repository at this point in the history
Co-authored-by: Roger Wang <[email protected]>
  • Loading branch information
haichuan1221 and ywang96 authored Jul 8, 2024
1 parent 16620f4 commit 717f4bc
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions benchmarks/benchmark_serving.py
Original file line number Diff line number Diff line change
Expand Up @@ -483,10 +483,10 @@ def main(args: argparse.Namespace):

elif args.dataset_name == "random":
input_requests = sample_random_requests(
input_len=args.input_len,
output_len=args.output_len,
input_len=args.random_input_len,
output_len=args.random_output_len,
num_prompts=args.num_prompts,
range_ratio=args.range_ratio,
range_ratio=args.random_range_ratio,
tokenizer=tokenizer,
)

Expand Down

0 comments on commit 717f4bc

Please sign in to comment.