A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 15, 13:30 UTC
[Misc] fix collect_env version parse (#15267)
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
wangxiyuan committed
Apr 21, 2025 at 20:29 UTC
d9ac9e3dc5a15a517d67a2c0f040ad0103e58251
1 file changed
+12
-5
vllm/collect_env.py
+12
-5
@@ -282,11 +282,18 @@ def get_vllm_version():
282
283
if __version__ == "dev":
284
return "N/A (dev)"
285
-
286
- if len(__version_tuple__) == 4: # dev build
287
- git_sha = __version_tuple__[-1][1:] # type: ignore
288
- return f"{__version__} (git sha: {git_sha}"
289
-
285
+ version_str = __version_tuple__[-1]
286
+ if isinstance(version_str, str) and version_str.startswith('g'):
287
+ # it's a dev build
288
+ if '.' in version_str:
289
+ # it's a dev build containing local changes
290
+ git_sha = version_str.split('.')[0][1:]
291
+ date = version_str.split('.')[-1][1:]
292
+ return f"{__version__} (git sha: {git_sha}, date: {date})"
293
+ else:
294
+ # it's a dev build without local changes
295
+ git_sha = version_str[1:] # type: ignore
296
+ return f"{__version__} (git sha: {git_sha})"
297
return __version__
298
299