Skip to content

Commit 5c03050

Browse files
committed
revert ipex import
Signed-off-by: Kaihui-intel <kaihui.tang@intel.com>
1 parent 783fde6 commit 5c03050

File tree

1 file changed

+4
-1
lines changed
  • neural_compressor/transformers/quantization

1 file changed

+4
-1
lines changed

neural_compressor/transformers/quantization/utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,10 @@
3535
convert,
3636
prepare,
3737
)
38-
from neural_compressor.torch.utils import is_package_available
38+
from neural_compressor.torch.utils import is_ipex_available, is_package_available
39+
40+
if is_ipex_available():
41+
import intel_extension_for_pytorch as ipex
3942

4043
if is_package_available("auto_round"):
4144
import auto_round

0 commit comments

Comments
 (0)