[Bugfix] Handle PackageNotFoundError when checking for xpu version (v…

…llm-project#7398)
neuralmagic · Aug 12, 2024 · 91294d5 · 91294d5
1 parent a046f86
commit 91294d5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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