diff --git a/vllm/benchmarks/datasets.py b/vllm/benchmarks/datasets.py index f06f41a47..86e080b55 100644 --- a/vllm/benchmarks/datasets.py +++ b/vllm/benchmarks/datasets.py @@ -1310,6 +1310,11 @@ class _ValidateDatasetArgs(argparse.Action): def add_dataset_parser(parser: FlexibleArgumentParser): + parser.add_argument( + "--trust-remote-code", + action="store_true", + help="Trust remote code from huggingface", + ) parser.add_argument("--seed", type=int, default=0) parser.add_argument( "--num-prompts", diff --git a/vllm/benchmarks/serve.py b/vllm/benchmarks/serve.py index 534392883..06e67f912 100644 --- a/vllm/benchmarks/serve.py +++ b/vllm/benchmarks/serve.py @@ -1313,11 +1313,6 @@ def add_cli_args(parser: argparse.ArgumentParser): "bursty requests. A higher burstiness value (burstiness > 1) " "results in a more uniform arrival of requests.", ) - parser.add_argument( - "--trust-remote-code", - action="store_true", - help="Trust remote code from huggingface", - ) parser.add_argument( "--disable-tqdm", action="store_true",