mirror of
https://github.com/wassname/vllm.git
synced 2026-09-13 13:10:13 +08:00
[BugFix] Use different mechanism to get vllm version in is_cpu() (#3804)
This commit is contained in:
+5
-3
@@ -119,9 +119,11 @@ def is_hip() -> bool:
|
||||
|
||||
@lru_cache(maxsize=None)
|
||||
def is_cpu() -> bool:
|
||||
from importlib.metadata import version
|
||||
is_cpu_flag = "cpu" in version("vllm")
|
||||
return is_cpu_flag
|
||||
from importlib.metadata import PackageNotFoundError, version
|
||||
try:
|
||||
return "cpu" in version("vllm")
|
||||
except PackageNotFoundError:
|
||||
return False
|
||||
|
||||
|
||||
@lru_cache(maxsize=None)
|
||||
|
||||
Reference in New Issue
Block a user