add served model name in bench serving (#12428)
This commit is contained in:
@@ -2275,7 +2275,7 @@ def run_benchmark(args_: argparse.Namespace):
|
|||||||
|
|
||||||
# Read dataset
|
# Read dataset
|
||||||
backend = args.backend
|
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_id = args.tokenizer if args.tokenizer is not None else args.model
|
||||||
tokenizer = get_tokenizer(tokenizer_id)
|
tokenizer = get_tokenizer(tokenizer_id)
|
||||||
input_requests = get_dataset(args, tokenizer, model_id)
|
input_requests = get_dataset(args, tokenizer, model_id)
|
||||||
@@ -2374,6 +2374,11 @@ if __name__ == "__main__":
|
|||||||
type=str,
|
type=str,
|
||||||
help="Name or path of the model. If not set, the default model will request /v1/models for conf.",
|
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(
|
parser.add_argument(
|
||||||
"--tokenizer",
|
"--tokenizer",
|
||||||
type=str,
|
type=str,
|
||||||
|
|||||||
Reference in New Issue
Block a user