Skip to content

Commit b3f7aac

Browse files
authored
[V1][Minor] Restore V1 compatibility with LLMEngine class (#13090)
1 parent b91660d commit b3f7aac

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
@@ -2084,3 +2084,8 @@ def _build_logits_processors(
20842084
sampling_params.logits_processors.extend(logits_processors)
20852085

20862086
return sampling_params
2087+
2088+
2089+
# TODO(v1): Remove this class proxy when V1 goes default.
2090+
if envs.VLLM_USE_V1:
2091+
from vllm.v1.engine.llm_engine import LLMEngine # type: ignore

0 commit comments

Comments
 (0)