From 3bcd494ef4bd50c8fa34990d80743728e464c2e0 Mon Sep 17 00:00:00 2001 From: Cyrus Leung Date: Wed, 11 Feb 2026 11:10:12 +0800 Subject: [PATCH] [Redo] Add `--trust-remote-code` to dataset bench args (#34251) Signed-off-by: DarkLight1337 --- vllm/benchmarks/datasets.py | 5 +++++ vllm/benchmarks/serve.py | 5 ----- 2 files changed, 5 insertions(+), 5 deletions(-) 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",