diff --git a/comfy/model_management.py b/comfy/model_management.py index c86b67e9f72..fd3979020b6 100644 --- 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)