[Bugfix] Handle PackageNotFoundError when checking for xpu version (#7398)

2024-08-12 23:07:20 +00:00 · 2024-08-12 23:07:20 +00:00 · 91294d56e1
commit 91294d56e1
parent a046f86397
1 changed files with 5 additions and 2 deletions
--- a/vllm/utils.py
+++ b/vllm/utils.py
@ -344,8 +344,11 @@ def is_tpu() -> bool:

@lru_cache(maxsize=None)
 def is_xpu() -> bool:
-    from importlib.metadata import version
-    is_xpu_flag = "xpu" in version("vllm")
+    from importlib.metadata import PackageNotFoundError, version
+    try:
+        is_xpu_flag = "xpu" in version("vllm")
+    except PackageNotFoundError:
+        return False
    # vllm is not build with xpu
    if not is_xpu_flag:
        return False