make the code more explicit

vllm-project · Sep 3, 2024 · dd4a388 · dd4a388
1 parent 534e75c
commit dd4a388
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -362,7 +362,7 @@ def get_vllm_version() -> str:
     version = find_version(get_path("vllm", "version.py"))
 
     if _no_device():
-        if VLLM_TARGET_DEVICE == envs.VLLM_TARGET_DEVICE:
+        if envs.VLLM_TARGET_DEVICE == "empty":
             version += "+empty"
     elif _is_cuda():
         cuda_version = str(get_nvcc_cuda_version())