Fix.

comfyanonymous · Aug 23, 2024 · 2c1d237 · 2c1d237
1 parent 64ccb3c
commit 2c1d237
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/comfy/model_management.py b/comfy/model_management.py
@@ -44,11 +44,15 @@ class CPUState(Enum):
 
 total_vram = 0
 
-torch_version = torch.version.__version__
+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()
+except:
+    pass
 
 lowvram_available = True
-xpu_available = int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)
-
 if args.deterministic:
     logging.info("Using deterministic algorithms for pytorch")
     torch.use_deterministic_algorithms(True, warn_only=True)