Update export function path in neural-compressor2.6

Signed-off-by: Cheng, Penghui <[email protected]>
PenghuiCheng · Jun 21, 2024 · 3898e37 · 3898e37
1 parent ef23a5d
commit 3898e37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/optimum/intel/neural_compressor/quantization.py b/optimum/intel/neural_compressor/quantization.py
@@ -25,7 +25,7 @@
 import torch
 from datasets import Dataset, load_dataset
 from neural_compressor.config import PostTrainingQuantConfig
-from neural_compressor.experimental.export import torch_to_int8_onnx
+from neural_compressor.utils.export import torch_to_int8_onnx
 from neural_compressor.model.onnx_model import ONNXModel
 from neural_compressor.model.torch_model import IPEXModel, PyTorchModel
 from neural_compressor.quantization import fit

diff --git a/optimum/intel/neural_compressor/trainer.py b/optimum/intel/neural_compressor/trainer.py
@@ -36,7 +36,7 @@
 from neural_compressor import training
 from neural_compressor.compression import DistillationCallbacks
 from neural_compressor.conf.pythonic_config import _BaseQuantizationConfig
-from neural_compressor.experimental.export import torch_to_fp32_onnx, torch_to_int8_onnx
+from neural_compressor.utils.export import torch_to_fp32_onnx, torch_to_int8_onnx
 from packaging import version
 from torch import nn
 from torch.utils.data import Dataset, RandomSampler