Skip to content

Commit 8691592

Browse files
committed
force set VLLM_USE_V1
Signed-off-by: Linkun Chen <github@lkchen.net>
1 parent 19362e9 commit 8691592

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/ray/llm/_internal/serve/deployments/llm/vllm/vllm_engine.py

+5
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,11 @@ async def _start_engine(self) -> "EngineClient":
315315
# 2. If VLLM_USE_V1 is set to 1, then it will use v1 engine even with
316316
# experimental features (such as launching vLLM on a non-main thread).
317317
# 3. If VLLM_USE_V1 is set to 0, force using v0 engine.
318+
if not envs.is_set("VLLM_USE_V1"):
319+
raise AssertionError(
320+
"Starting ray 2.45, VLLM_USE_V1 environmetn variable must be set "
321+
"to prevent undetermined behavior"
322+
)
318323
if not envs.VLLM_USE_V1:
319324
return await self._start_engine_v0()
320325
return await self._start_engine_v1()

0 commit comments

Comments
 (0)