xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-23 15:24:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-23 04:06:27 -04:00
parent 103d705247
commit 65a577e0db
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -47,8 +47,7 @@ total_vram = 0
xpu_available = False xpu_available = False
try: try:
torch_version = torch.version.__version__ 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 = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available()
xpu_available = xpu_available and torch.xpu.is_available()
except: except:
pass pass