xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 09:15:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Frontend] Add --version flag to CLI (#10369)

Browse Source 
Signed-off-by: Russell Bryant <rbryant@redhat.com>
This commit is contained in:
Russell Bryant 2024-11-15 16:13:53 -05:00 committed by GitHub
parent c76ac49d26
commit a067f85e08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/scripts.py
View File

@ -9,6 +9,7 @@ import uvloop
from openai import OpenAI from openai import OpenAI
from openai.types.chat import ChatCompletionMessageParam from openai.types.chat import ChatCompletionMessageParam
import vllm.version
from vllm.engine.arg_utils import EngineArgs from vllm.engine.arg_utils import EngineArgs
from vllm.entrypoints.openai.api_server import run_server from vllm.entrypoints.openai.api_server import run_server
from vllm.entrypoints.openai.cli_args import (make_arg_parser, from vllm.entrypoints.openai.cli_args import (make_arg_parser,
@ -143,6 +144,11 @@ def main():
env_setup() env_setup()
parser = FlexibleArgumentParser(description="vLLM CLI") parser = FlexibleArgumentParser(description="vLLM CLI")
parser.add_argument('-v',
'--version',
action='version',
version=vllm.version.__version__)
subparsers = parser.add_subparsers(required=True, dest="subparser") subparsers = parser.add_subparsers(required=True, dest="subparser")
serve_parser = subparsers.add_parser( serve_parser = subparsers.add_parser(