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

This commit is contained in:
youkaichao 2024-08-11 15:40:48 -07:00 committed by GitHub
parent 02b1988b9f
commit 6c8e595710
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -265,8 +265,9 @@ def get_neuron_sdk_version(run_lambda):
def get_vllm_version():
try:
import vllm
return vllm.__version__
except ImportError:
return vllm.__version__ + "@" + vllm.__commit__
except Exception:
# old version of vllm does not have __commit__
return 'N/A'