From af985d70bfa40bda726ce369287c9be0311ae33e Mon Sep 17 00:00:00 2001 From: LiuXiaoxuanPKU Date: Tue, 1 Apr 2025 15:53:26 -0700 Subject: [PATCH] change to greedy --- benchmarks/benchmark_latency.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/benchmark_latency.py b/benchmarks/benchmark_latency.py index e03c724baa..f0647d8370 100644 --- a/benchmarks/benchmark_latency.py +++ b/benchmarks/benchmark_latency.py @@ -51,7 +51,7 @@ def main(args: argparse.Namespace): sampling_params = SamplingParams( n=args.n, - temperature=1.0, + temperature=0, top_p=1.0, ignore_eos=True, max_tokens=args.output_len,