Cleanup.

6 months ago · 5d8bbb7281
parent 2c1d2375d6
commit 5d8bbb7281
1 changed files with 1 additions and 2 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -47,8 +47,7 @@ total_vram = 0
 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