[FD CLI] Add bench cli (#4160)

* add bench cli

* Update test_main.py
This commit is contained in:
Zhang Yulong
2025-09-22 20:37:30 +08:00
committed by GitHub
parent 5e1f13bd3b
commit 5532e8a323
11 changed files with 2890 additions and 3 deletions

View File

@@ -21,11 +21,13 @@ import importlib.metadata
def main():
import fastdeploy.entrypoints.cli.benchmark.main
import fastdeploy.entrypoints.cli.openai
from fastdeploy.utils import FlexibleArgumentParser
CMD_MODULES = [
fastdeploy.entrypoints.cli.openai,
fastdeploy.entrypoints.cli.benchmark.main,
]
parser = FlexibleArgumentParser(description="FastDeploy CLI")
@@ -33,7 +35,7 @@ def main():
"-v",
"--version",
action="version",
version=importlib.metadata.version("fastdeploy"),
version=importlib.metadata.version("fastdeploy-gpu"),
)
subparsers = parser.add_subparsers(required=False, dest="subparser")
cmds = {}