Initialize AsyncLLMEngine bg loop correctly (#943)

This commit is contained in:
Antoni Baum
2023-09-04 17:41:22 -07:00
committed by GitHub
parent 002800f081
commit 1696725879
3 changed files with 25 additions and 7 deletions

View File

@@ -30,6 +30,10 @@ async def generate(request: Request) -> Response:
stream = request_dict.pop("stream", False)
sampling_params = SamplingParams(**request_dict)
request_id = random_uuid()
if not engine.is_running:
engine.start_background_loop()
results_generator = engine.generate(prompt, sampling_params, request_id)
# Streaming case
@@ -75,7 +79,8 @@ if __name__ == "__main__":
args = parser.parse_args()
engine_args = AsyncEngineArgs.from_cli_args(args)
engine = AsyncLLMEngine.from_engine_args(engine_args)
engine = AsyncLLMEngine.from_engine_args(engine_args,
start_engine_loop=False)
uvicorn.run(app,
host=args.host,