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