frozenleaves/vllm
1
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2025-11-04 17:34:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Start background task in AsyncLLMEngine.generate (#988)

Browse Source 
Co-authored-by: Zhuohan Li <zhuohan123@gmail.com>
This commit is contained in:
Antoni Baum
2023-09-08 00:03:39 -07:00
committed by GitHub
parent 4b5bcf8906
commit 080438477f
4 changed files with 16 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/async_engine/api_server_async_engine.py
View File

@ -40,8 +40,7 @@ if __name__ == "__main__":
args = parser.parse_args()
engine_args = AsyncEngineArgs.from_cli_args(args)
engine = AsyncLLMEngineWithStats.from_engine_args(engine_args,
start_engine_loop=False)
engine = AsyncLLMEngineWithStats.from_engine_args(engine_args)
vllm.entrypoints.api_server.engine = engine
uvicorn.run(
app,