Skip to content

Commit e879a5a

Browse files
evaline-jujoerunde
authored andcommitted
👽 Update for vllm config
Signed-off-by: Evaline Ju <[email protected]>
1 parent 09d5964 commit e879a5a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/vllm_tgis_adapter/http.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,12 @@ async def set_correlation_id(request: Request, call_next: Callable) -> Response:
4646

4747
return await call_next(request)
4848

49-
model_config = await engine.get_model_config()
50-
maybe_coroutine = init_app_state(engine, model_config, app.state, args)
49+
if hasattr(engine, "get_vllm_config"):
50+
vllm_config = await engine.get_vllm_config()
51+
maybe_coroutine = init_app_state(engine, vllm_config, app.state, args)
52+
else:
53+
model_config = await engine.get_model_config()
54+
maybe_coroutine = init_app_state(engine, model_config, app.state, args)
5155
if inspect.isawaitable(maybe_coroutine):
5256
await maybe_coroutine
5357

0 commit comments

Comments
 (0)