Skip to content

Commit 7274d97

Browse files
Fix imports 2
1 parent 2f211fc commit 7274d97

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

optimum/intel/openvino/quantization/quantizer.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,10 @@
1919
from pathlib import Path
2020
from typing import Callable, Dict, Iterable, List, Optional, Sized, Union
2121

22-
import nncf
2322
import openvino
2423
import torch
2524
import transformers
2625
from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE
27-
from nncf.quantization.advanced_parameters import OverflowFix
28-
from nncf.torch import register_module
2926
from openvino._offline_transformations import compress_quantize_weights_transformation
3027
from openvino.runtime import Core
3128
from torch.utils._pytree import tree_map
@@ -41,7 +38,7 @@
4138
from ...utils.constant import _TASK_ALIASES
4239
from ...utils.import_utils import (
4340
is_datasets_available,
44-
is_diffusers_available,
41+
is_diffusers_available, is_nncf_available,
4542
)
4643
from ...utils.modeling_utils import get_model_device
4744
from ..modeling_base import OVBaseModel
@@ -64,6 +61,11 @@
6461
if is_datasets_available():
6562
from datasets import Dataset
6663

64+
if is_nncf_available():
65+
import nncf
66+
from nncf.quantization.advanced_parameters import OverflowFix
67+
from nncf.torch import register_module
68+
6769
register_module(ignored_algorithms=[])(Conv1D)
6870

6971
core = Core()

0 commit comments

Comments
 (0)