Skip to content

Commit 5739371

Browse files
authored
[Misc] VLLM_TARGET_DEVICE.lower() (#22101)
Signed-off-by: NickLucche <[email protected]>
1 parent ee2eb6e commit 5739371

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/envs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def get_vllm_port() -> Optional[int]:
213213
# Target device of vLLM, supporting [cuda (by default),
214214
# rocm, neuron, cpu]
215215
"VLLM_TARGET_DEVICE":
216-
lambda: os.getenv("VLLM_TARGET_DEVICE", "cuda"),
216+
lambda: os.getenv("VLLM_TARGET_DEVICE", "cuda").lower(),
217217

218218
# Maximum number of compilation jobs to run in parallel.
219219
# By default this is the number of CPUs

0 commit comments

Comments
 (0)