add served model name in bench serving (#12428)

This commit is contained in:
daniel, chen
2025-11-01 01:28:11 +08:00
committed by GitHub
parent 5538e05cb1
commit cf0c24150a

View File

@@ -2275,7 +2275,7 @@ def run_benchmark(args_: argparse.Namespace):
# Read dataset
backend = args.backend
model_id = args.model
model_id = args.served_model_name or args.model
tokenizer_id = args.tokenizer if args.tokenizer is not None else args.model
tokenizer = get_tokenizer(tokenizer_id)
input_requests = get_dataset(args, tokenizer, model_id)
@@ -2374,6 +2374,11 @@ if __name__ == "__main__":
type=str,
help="Name or path of the model. If not set, the default model will request /v1/models for conf.",
)
parser.add_argument(
"--served-model-name",
type=str,
help="The name of the model as served by the serving service. If not set, this defaults to the value of --model.",
)
parser.add_argument(
"--tokenizer",
type=str,