mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-12 16:26:07 +08:00
71b1c8b667
Signed-off-by: Yeshwanth Surya <yeshsurya@gmail.com> Signed-off-by: Yeshwanth N <yeshsurya@gmail.com> Signed-off-by: yeshsurya <yeshsurya@gmail.com>
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""The CLI entrypoints of vLLM
|
|
|
|
Note that all future modules must be lazily loaded within main
|
|
to avoid certain eager import breakage."""
|
|
|
|
import importlib.metadata
|
|
import sys
|
|
|
|
from vllm.logger import init_logger
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
def main():
|
|
import vllm.entrypoints.cli.benchmark.main
|
|
import vllm.entrypoints.cli.collect_env
|
|
import vllm.entrypoints.cli.openai
|
|
import vllm.entrypoints.cli.run_batch
|
|
import vllm.entrypoints.cli.serve
|
|
from vllm.entrypoints.utils import VLLM_SUBCMD_PARSER_EPILOG, cli_env_setup
|
|
from vllm.utils.argparse_utils import FlexibleArgumentParser
|
|
|
|
CMD_MODULES = [
|
|
vllm.entrypoints.cli.openai,
|
|
vllm.entrypoints.cli.serve,
|
|
vllm.entrypoints.cli.benchmark.main,
|
|
vllm.entrypoints.cli.collect_env,
|
|
vllm.entrypoints.cli.run_batch,
|
|
]
|
|
|
|
cli_env_setup()
|
|
|
|
# For 'vllm bench *': use CPU instead of UnspecifiedPlatform by default
|
|
if len(sys.argv) > 1 and sys.argv[1] == "bench":
|
|
logger.debug(
|
|
"Bench command detected, must ensure current platform is not "
|
|
"UnspecifiedPlatform to avoid device type inference error"
|
|
)
|
|
from vllm import platforms
|
|
|
|
if platforms.current_platform.is_unspecified():
|
|
from vllm.platforms.cpu import CpuPlatform
|
|
|
|
platforms.current_platform = CpuPlatform()
|
|
logger.info(
|
|
"Unspecified platform detected, switching to CPU Platform instead."
|
|
)
|
|
|
|
parser = FlexibleArgumentParser(
|
|
description="vLLM CLI",
|
|
epilog=VLLM_SUBCMD_PARSER_EPILOG.format(subcmd="[subcommand]"),
|
|
)
|
|
parser.add_argument(
|
|
"-v",
|
|
"--version",
|
|
action="version",
|
|
version=importlib.metadata.version("vllm"),
|
|
)
|
|
subparsers = parser.add_subparsers(required=False, dest="subparser")
|
|
cmds = {}
|
|
for cmd_module in CMD_MODULES:
|
|
new_cmds = cmd_module.cmd_init()
|
|
for cmd in new_cmds:
|
|
cmd.subparser_init(subparsers).set_defaults(dispatch_function=cmd.cmd)
|
|
cmds[cmd.name] = cmd
|
|
args = parser.parse_args()
|
|
if args.subparser in cmds:
|
|
cmds[args.subparser].validate(args)
|
|
|
|
if hasattr(args, "dispatch_function"):
|
|
args.dispatch_function(args)
|
|
else:
|
|
parser.print_help()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|