Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
dc4e3df5
Unverified
Commit
dc4e3df5
authored
Sep 27, 2024
by
youkaichao
Committed by
GitHub
Sep 27, 2024
Browse files
[misc] fix collect env (#8894)
parent
3b00b9c2
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
4 deletions
+14
-4
collect_env.py
collect_env.py
+14
-4
No files found.
collect_env.py
View file @
dc4e3df5
...
@@ -267,13 +267,23 @@ def get_neuron_sdk_version(run_lambda):
...
@@ -267,13 +267,23 @@ def get_neuron_sdk_version(run_lambda):
def
get_vllm_version
():
def
get_vllm_version
():
version
=
""
try
:
try
:
import
vllm
import
vllm
return
vllm
.
__version__
+
"@"
+
vllm
.
__commit__
version
=
vllm
.
__version__
except
Exception
:
except
Exception
:
# old version of vllm does not have __commit__
pass
return
'N/A'
commit
=
""
try
:
import
vllm
commit
=
vllm
.
__commit__
except
Exception
:
pass
if
version
!=
""
and
commit
!=
""
:
return
f
"
{
version
}
@
{
commit
}
"
if
version
==
""
and
commit
==
""
:
return
"N/A"
return
version
or
commit
def
summarize_vllm_build_flags
():
def
summarize_vllm_build_flags
():
# This could be a static method if the flags are constant, or dynamic if you need to check environment variables, etc.
# This could be a static method if the flags are constant, or dynamic if you need to check environment variables, etc.
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment