Import intel_extension_for_pytorch as ipex

This commit is contained in:
藍+85CD 2023-04-06 12:27:22 +08:00
parent a5b1eb2936
commit 772741f218

View File

@ -87,7 +87,7 @@ except:
pass
try:
import intel_extension_for_pytorch
import intel_extension_for_pytorch as ipex
if torch.xpu.is_available():
vram_state = XPU
except: