Skip to content

GitLab

Unverified Commit 6c8e5957 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[misc] add commit id in collect env (#7405)

parent 02b1988b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
collect_env.py
View file @ 6c8e5957
...@@ -265,8 +265,9 @@ def get_neuron_sdk_version(run_lambda): ...@@ -265,8 +265,9 @@ def get_neuron_sdk_version(run_lambda):
def get_vllm_version(): def get_vllm_version():
try: try:
import vllm import vllm
return vllm.__version__ return vllm.__version__ + "@" + vllm.__commit__
except ImportError: except Exception:
# old version of vllm does not have __commit__
return 'N/A' return 'N/A'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment