xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 06:53:31 +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 ac9a828f50
commit 749b5f2a28
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -68,6 +68,8 @@ try:
import intel_extension_for_pytorch as ipex import intel_extension_for_pytorch as ipex
if torch.xpu.is_available(): if torch.xpu.is_available():
xpu_available = True xpu_available = True
from ipex_hijacks import hijacks
hijacks.ipex_hijacks()
except: except:
pass pass