Skip to content

Commit 4a6f4bd

Browse files
committed
[V1][Minor] Restore V1 compatibility with LLMEngine class
The check was added to AsyncLLMEngine but not LLMEngine. Signed-off-by: Thibault Schueller <[email protected]> Signed-off-by: 'Thibault Schueller' <'[email protected]'>
1 parent 7ca1da0 commit 4a6f4bd

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

vllm/engine/llm_engine.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2028,3 +2028,8 @@ def _build_logits_processors(
20282028
sampling_params.logits_processors.extend(logits_processors)
20292029

20302030
return sampling_params
2031+
2032+
2033+
# TODO(v1): Remove this class proxy when V1 goes default.
2034+
if envs.VLLM_USE_V1:
2035+
from vllm.v1.engine.llm_engine import LLMEngine # type: ignore

0 commit comments

Comments
 (0)