diff --git a/comfy/model_management.py b/comfy/model_management.py index 6b031e204..19e57a79d 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -45,11 +45,12 @@ cpu_state = CPUState.GPU total_vram = 0 xpu_available = False #We don't have xpu with zluda , so disabling all together. -# 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)) and torch.xpu.is_available() -# except: -# pass +torch_version = "" + 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)) and torch.xpu.is_available() + except: + pass lowvram_available = True if args.deterministic: