Skip to content

Commit 3898e37

Browse files
committedJun 21, 2024
Update export function path in neural-compressor2.6
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent ef23a5d commit 3898e37

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed
 

‎optimum/intel/neural_compressor/quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
import torch
2626
from datasets import Dataset, load_dataset
2727
from neural_compressor.config import PostTrainingQuantConfig
28-
from neural_compressor.experimental.export import torch_to_int8_onnx
28+
from neural_compressor.utils.export import torch_to_int8_onnx
2929
from neural_compressor.model.onnx_model import ONNXModel
3030
from neural_compressor.model.torch_model import IPEXModel, PyTorchModel
3131
from neural_compressor.quantization import fit

‎optimum/intel/neural_compressor/trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
from neural_compressor import training
3737
from neural_compressor.compression import DistillationCallbacks
3838
from neural_compressor.conf.pythonic_config import _BaseQuantizationConfig
39-
from neural_compressor.experimental.export import torch_to_fp32_onnx, torch_to_int8_onnx
39+
from neural_compressor.utils.export import torch_to_fp32_onnx, torch_to_int8_onnx
4040
from packaging import version
4141
from torch import nn
4242
from torch.utils.data import Dataset, RandomSampler

0 commit comments

Comments
 (0)