diff --git a/comfy/model_management.py b/comfy/model_management.py index fd397902..91e692ba 100644 --- 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