Skip to content

Commit 2f211fc

Browse files
Fix imports
1 parent 6f55441 commit 2f211fc

File tree

5 files changed

+8
-5
lines changed

5 files changed

+8
-5
lines changed

notebooks/openvino/stable_diffusion_hybrid_quantization.ipynb

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
"from pathlib import Path\n",
5454
"from openvino.runtime import Core\n",
5555
"from optimum.intel import OVConfig, OVQuantizer, OVStableDiffusionPipeline, OVWeightQuantizationConfig\n",
56-
"from optimum.intel.openvino.quantization.configuration import OVQuantizationMethod\n",
56+
"from optimum.intel.openvino.quantization import OVQuantizationMethod\n",
5757
"\n",
5858
"transformers.logging.set_verbosity_error()\n",
5959
"datasets.logging.set_verbosity_error()"

optimum/exporters/openvino/__main__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@
5252
FORCE_ATTN_MODEL_CLASSES = {"phi3-v": "eager", "gemma2": "sdpa"}
5353

5454
if TYPE_CHECKING:
55-
from optimum.intel.openvino.quantization.configuration import OVConfig
55+
from optimum.intel.openvino.quantization import OVConfig
5656

5757

5858
if is_torch_available():

optimum/exporters/openvino/convert.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@
9090

9191
if TYPE_CHECKING:
9292
from optimum.exporters.onnx.base import OnnxConfig
93-
from optimum.intel.openvino.quantization.configuration import OVConfig
93+
from optimum.intel.openvino.quantization import OVConfig
9494

9595

9696
def _set_runtime_options(

optimum/intel/openvino/configuration.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,6 @@
3333
logger = logging.getLogger(__name__)
3434

3535
logger.warning(
36-
"`optimum.intel.configuration` import path is deprecated and will be removed in optimum-intel v.1.26."
36+
"`optimum.intel.configuration` import path is deprecated and will be removed in optimum-intel v1.25."
3737
"Please use `optimum.intel.quantization.configuration` instead."
3838
)

optimum/intel/openvino/quantization/calibration_dataset_builder.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
import logging
1717
from typing import Any, Callable, Dict, List, Optional, Sized, Tuple, Union
1818

19-
import nncf
2019
import openvino
2120
import requests
2221
import torch
@@ -40,6 +39,7 @@
4039
is_datasets_available,
4140
is_datasets_version,
4241
is_diffusers_available,
42+
is_nncf_available,
4343
)
4444

4545
from .configuration import OVQuantizationConfigBase
@@ -48,6 +48,9 @@
4848
if is_datasets_available():
4949
from datasets import Dataset
5050

51+
if is_nncf_available():
52+
import nncf
53+
5154
logger = logging.getLogger(__name__)
5255

5356

0 commit comments

Comments
 (0)