Update offline_transformations (#140)

huggingface · Dec 24, 2022 · 226bc96 · 226bc96
1 parent dc6f2a2
commit 226bc96
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py
@@ -26,7 +26,7 @@
 import openvino
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import EntryNotFoundError
-from openvino.offline_transformations import apply_moc_transformations, compress_model_transformation
+from openvino._offline_transformations import apply_moc_transformations, compress_model_transformation
 from openvino.runtime import Core
 from optimum.modeling_base import OptimizedModel
 

diff --git a/optimum/intel/openvino/modeling_base_seq2seq.py b/optimum/intel/openvino/modeling_base_seq2seq.py
@@ -26,7 +26,7 @@
 import openvino
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.utils import EntryNotFoundError
-from openvino.offline_transformations import compress_model_transformation
+from openvino._offline_transformations import compress_model_transformation
 from optimum.onnx.configuration import DecoderOnnxConfig, EncoderOnnxConfig
 from optimum.onnx.modeling_seq2seq import _DecoderWithLMhead
 

diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py
@@ -35,7 +35,7 @@
 from nncf.torch.dynamic_graph.io_handling import wrap_nncf_model_inputs_with_objwalk
 from nncf.torch.initialization import PTInitializingDataLoader
 from nncf.torch.nncf_network import NNCFNetwork
-from openvino.offline_transformations import compress_quantize_weights_transformation
+from openvino._offline_transformations import compress_quantize_weights_transformation
 from openvino.runtime import Core
 from optimum.quantization_base import OptimumQuantizer
 

diff --git a/optimum/intel/openvino/trainer.py b/optimum/intel/openvino/trainer.py
@@ -58,7 +58,7 @@
 from nncf.config.structures import BNAdaptationInitArgs, QuantizationRangeInitArgs
 from nncf.torch import create_compressed_model
 from nncf.torch.nncf_network import NNCFNetwork
-from openvino.offline_transformations import compress_quantize_weights_transformation
+from openvino._offline_transformations import compress_quantize_weights_transformation
 from openvino.runtime import Core
 from optimum.utils import logging