From 749b5f2a286e75f8f9da5eb5dcb1c2124e24b503 Mon Sep 17 00:00:00 2001 From: Kel Phillipson Date: Fri, 16 Aug 2024 11:53:49 -0700 Subject: [PATCH] Call ipex hijacks in model management --- comfy/model_management.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/comfy/model_management.py b/comfy/model_management.py index a6996709b..b3b8a394e 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -68,6 +68,8 @@ try: import intel_extension_for_pytorch as ipex if torch.xpu.is_available(): xpu_available = True + from ipex_hijacks import hijacks + hijacks.ipex_hijacks() except: pass