mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-21 21:15:56 +08:00
[bugfix] fix the wrong parser (#17958)
Signed-off-by: reidliu41 <reid201711@gmail.com> Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
parent
08bf784078
commit
ada50aa295
@ -4,12 +4,11 @@ import argparse
|
|||||||
|
|
||||||
from vllm.collect_env import main as collect_env_main
|
from vllm.collect_env import main as collect_env_main
|
||||||
from vllm.entrypoints.cli.types import CLISubcommand
|
from vllm.entrypoints.cli.types import CLISubcommand
|
||||||
from vllm.entrypoints.openai.cli_args import make_arg_parser
|
|
||||||
from vllm.utils import FlexibleArgumentParser
|
from vllm.utils import FlexibleArgumentParser
|
||||||
|
|
||||||
|
|
||||||
class CollectEnvSubcommand(CLISubcommand):
|
class CollectEnvSubcommand(CLISubcommand):
|
||||||
"""The `serve` subcommand for the vLLM CLI. """
|
"""The `collect-env` subcommand for the vLLM CLI. """
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.name = "collect-env"
|
self.name = "collect-env"
|
||||||
@ -23,12 +22,12 @@ class CollectEnvSubcommand(CLISubcommand):
|
|||||||
def subparser_init(
|
def subparser_init(
|
||||||
self,
|
self,
|
||||||
subparsers: argparse._SubParsersAction) -> FlexibleArgumentParser:
|
subparsers: argparse._SubParsersAction) -> FlexibleArgumentParser:
|
||||||
serve_parser = subparsers.add_parser(
|
collect_env_parser = subparsers.add_parser(
|
||||||
"collect-env",
|
"collect-env",
|
||||||
help="Start collecting environment information.",
|
help="Start collecting environment information.",
|
||||||
description="Start collecting environment information.",
|
description="Start collecting environment information.",
|
||||||
usage="vllm collect-env")
|
usage="vllm collect-env")
|
||||||
return make_arg_parser(serve_parser)
|
return collect_env_parser
|
||||||
|
|
||||||
|
|
||||||
def cmd_init() -> list[CLISubcommand]:
|
def cmd_init() -> list[CLISubcommand]:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user