[Misc] Add vLLM version getter to utils (#5098)

2024-06-14 02:21:39 +08:00
parent a65634d3ae
commit 03dccc886e
7 changed files with 13 additions and 11 deletions
--- a/setup.py
+++ b/setup.py
@@ -314,7 +314,7 @@ def find_version(filepath: str) -> str:


 def get_vllm_version() -> str:
-    version = find_version(get_path("vllm", "__init__.py"))
+    version = find_version(get_path("vllm", "version.py"))

    if _is_cuda():
        cuda_version = str(get_nvcc_cuda_version())