From 8da165ec22a555e94e0a3cf9a3f1c0a1b26c4ab8 Mon Sep 17 00:00:00 2001
From: Kel Phillipson <kel@kelfish.com>
Date: Fri, 16 Aug 2024 11:53:49 -0700
Subject: [PATCH] Call ipex hijacks in model management

---
 comfy/model_management.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 77c05510e..92c9031d7 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -77,6 +77,10 @@ try:
 except:
     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:
     if torch.backends.mps.is_available():
         cpu_state = CPUState.MPS