Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
6885fde3
Unverified
Commit
6885fde3
authored
Aug 23, 2024
by
Pooya Davoodi
Committed by
GitHub
Aug 23, 2024
Browse files
[Bugfix] Fix run_batch logger (#7640)
parent
9db93de2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
5 deletions
+2
-5
vllm/entrypoints/openai/run_batch.py
vllm/entrypoints/openai/run_batch.py
+2
-5
No files found.
vllm/entrypoints/openai/run_batch.py
View file @
6885fde3
...
@@ -6,7 +6,7 @@ import aiohttp
...
@@ -6,7 +6,7 @@ import aiohttp
from
vllm.engine.arg_utils
import
AsyncEngineArgs
,
nullable_str
from
vllm.engine.arg_utils
import
AsyncEngineArgs
,
nullable_str
from
vllm.engine.async_llm_engine
import
AsyncLLMEngine
from
vllm.engine.async_llm_engine
import
AsyncLLMEngine
from
vllm.entrypoints.logger
import
RequestLogger
from
vllm.entrypoints.logger
import
RequestLogger
,
logger
# yapf: disable
# yapf: disable
from
vllm.entrypoints.openai.protocol
import
(
BatchRequestInput
,
from
vllm.entrypoints.openai.protocol
import
(
BatchRequestInput
,
BatchRequestOutput
,
BatchRequestOutput
,
...
@@ -16,13 +16,10 @@ from vllm.entrypoints.openai.protocol import (BatchRequestInput,
...
@@ -16,13 +16,10 @@ from vllm.entrypoints.openai.protocol import (BatchRequestInput,
# yapf: enable
# yapf: enable
from
vllm.entrypoints.openai.serving_chat
import
OpenAIServingChat
from
vllm.entrypoints.openai.serving_chat
import
OpenAIServingChat
from
vllm.entrypoints.openai.serving_embedding
import
OpenAIServingEmbedding
from
vllm.entrypoints.openai.serving_embedding
import
OpenAIServingEmbedding
from
vllm.logger
import
init_logger
from
vllm.usage.usage_lib
import
UsageContext
from
vllm.usage.usage_lib
import
UsageContext
from
vllm.utils
import
FlexibleArgumentParser
,
random_uuid
from
vllm.utils
import
FlexibleArgumentParser
,
random_uuid
from
vllm.version
import
__version__
as
VLLM_VERSION
from
vllm.version
import
__version__
as
VLLM_VERSION
logger
=
init_logger
(
__name__
)
def
parse_args
():
def
parse_args
():
parser
=
FlexibleArgumentParser
(
parser
=
FlexibleArgumentParser
(
...
@@ -184,7 +181,7 @@ async def main(args):
...
@@ -184,7 +181,7 @@ async def main(args):
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
args
=
parse_args
()
args
=
parse_args
()
logger
.
info
(
"vLLM
API server
version %s"
,
VLLM_VERSION
)
logger
.
info
(
"vLLM
batch processing API
version %s"
,
VLLM_VERSION
)
logger
.
info
(
"args: %s"
,
args
)
logger
.
info
(
"args: %s"
,
args
)
asyncio
.
run
(
main
(
args
))
asyncio
.
run
(
main
(
args
))
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment