Cleanup.

Comfy-Org · Aug 23, 2024 · 5d8bbb7 · 5d8bbb7
1 parent 2c1d237
commit 5d8bbb7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -47,8 +47,7 @@ class CPUState(Enum):
 xpu_available = False
 try:
     torch_version = torch.version.__version__
-    xpu_available = int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)
-    xpu_available = xpu_available and torch.xpu.is_available()
+    xpu_available = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available()
 except:
     pass