Unverified Commit cf0c2415 authored by carolove's avatar carolove Committed by GitHub
Browse files

add served model name in bench serving (#12428)

parent 5538e05c
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment