xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-13 02:47:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Call ipex hijacks in model management

Browse Source
This commit is contained in:
Kel Phillipson 2024-08-16 11:53:49 -07:00
parent 3586895584
commit 8da165ec22
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -77,6 +77,10 @@ try:
except: except:
xpu_available = xpu_available or (hasattr(torch, "xpu") and torch.xpu.is_available()) xpu_available = xpu_available or (hasattr(torch, "xpu") and torch.xpu.is_available())
if xpu_available:
from ipex_hijacks import hijacks
hijacks.ipex_hijacks()
try: try:
if torch.backends.mps.is_available(): if torch.backends.mps.is_available():
cpu_state = CPUState.MPS cpu_state = CPUState.MPS