Skip to content

Commit c9b506d

Browse files
authored
[BugFix] Use different mechanism to get vllm version in is_cpu() (#3804)
1 parent 5757d90 commit c9b506d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

vllm/utils.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -119,9 +119,11 @@ def is_hip() -> bool:
119119

120120
@lru_cache(maxsize=None)
121121
def is_cpu() -> bool:
122-
from importlib.metadata import version
123-
is_cpu_flag = "cpu" in version("vllm")
124-
return is_cpu_flag
122+
from importlib.metadata import PackageNotFoundError, version
123+
try:
124+
return "cpu" in version("vllm")
125+
except PackageNotFoundError:
126+
return False
125127

126128

127129
@lru_cache(maxsize=None)

0 commit comments

Comments
 (0)