298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-23 04:04:55 -04:00
parent 64ccb3c7e3
commit 2c1d2375d6
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/model_management.py
View File

@ -44,11 +44,15 @@ cpu_state = CPUState.GPU
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)