Skip to content

Commit d544d14

Browse files
authored
update benchmark_serving_structured_output to include auto backend (#16438)
Signed-off-by: Chenyaaang <[email protected]>
1 parent 3e397a9 commit d544d14

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

benchmarks/benchmark_serving_structured_output.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
--model <your_model> \
1212
--dataset json \
1313
--structured-output-ratio 1.0 \
14-
--structured-output-backend xgrammar \
14+
--structured-output-backend auto \
1515
--request-rate 10 \
1616
--num-prompts 1000
1717
@@ -997,12 +997,14 @@ def main(args: argparse.Namespace):
997997
type=float,
998998
default=1.0,
999999
help="Ratio of Structured Outputs requests")
1000-
parser.add_argument(
1001-
"--structured-output-backend",
1002-
type=str,
1003-
choices=["outlines", "lm-format-enforcer", "xgrammar", "guidance"],
1004-
default="xgrammar",
1005-
help="Backend to use for structured outputs")
1000+
parser.add_argument("--structured-output-backend",
1001+
type=str,
1002+
choices=[
1003+
"outlines", "lm-format-enforcer", "xgrammar",
1004+
"guidance", "auto"
1005+
],
1006+
default="auto",
1007+
help="Backend to use for structured outputs")
10061008

10071009
args = parser.parse_args()
10081010
main(args)

0 commit comments

Comments
 (0)