Commit 84b9c0ac authored by 藍+85CD's avatar 藍+85CD
Browse files

Import intel_extension_for_pytorch as ipex

parent 1ced2bdd
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment