diff --git a/benchmarks/benchmark_serving.py b/benchmarks/benchmark_serving.py index 0624afe7dc..bfe8b02701 100644 --- a/benchmarks/benchmark_serving.py +++ b/benchmarks/benchmark_serving.py @@ -1123,8 +1123,10 @@ def save_to_pytorch_benchmark_format(args: argparse.Namespace, results: dict[str def main(args: argparse.Namespace): """Main entry point""" print(args) - random.seed(args.seed) - np.random.seed(args.seed) + if args.seed is not None: + print(f"Using random seed: {args.seed}") + random.seed(args.seed) + np.random.seed(args.seed) backend = args.backend # 支持多轮对话方式请求,仅支持chat接口 @@ -1431,7 +1433,7 @@ if __name__ == "__main__": "bursty requests. A higher burstiness value (burstiness > 1) " "results in a more uniform arrival of requests.", ) - parser.add_argument("--seed", type=int, default=0) + parser.add_argument("--seed", type=int, default=None) parser.add_argument( "--shuffle", action="store_true",