Call ipex hijacks in model management

This commit is contained in:
Kel Phillipson 2024-08-16 11:53:49 -07:00
parent ac9a828f50
commit 749b5f2a28

View File

@ -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