Skip to content

Commit b65b313

Browse files
Fix imports 6
1 parent 36a56b9 commit b65b313

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

optimum/intel/openvino/quantization/calibration_dataset_builder.py

+2-5
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,13 @@
1616
import logging
1717
from typing import Any, Callable, Dict, List, Optional, Sized, Tuple, Union
1818

19+
import nncf
1920
import openvino
2021
import requests
2122
import torch
2223
import transformers
2324
from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE
25+
from nncf.torch.initialization import PTInitializingDataLoader
2426
from PIL import Image
2527
from torch.utils.data import DataLoader, RandomSampler
2628
from tqdm import tqdm
@@ -37,7 +39,6 @@
3739
is_datasets_available,
3840
is_datasets_version,
3941
is_diffusers_available,
40-
is_nncf_available,
4142
)
4243

4344
from .configuration import OVQuantizationConfigBase
@@ -46,10 +47,6 @@
4647
if is_datasets_available():
4748
from datasets import Dataset
4849

49-
if is_nncf_available():
50-
import nncf
51-
from nncf.torch.initialization import PTInitializingDataLoader
52-
5350
logger = logging.getLogger(__name__)
5451

5552

optimum/intel/openvino/quantization/quantizer.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@
4848
ONNX_WEIGHTS_NAME,
4949
OV_XML_FILE_NAME,
5050
)
51-
from .calibration_dataset_builder import OVCalibrationDatasetBuilder
5251
from .configuration import (
5352
OVConfig,
5453
OVMixedQuantizationConfig,
@@ -66,6 +65,8 @@
6665
from nncf.quantization.advanced_parameters import OverflowFix
6766
from nncf.torch import register_module
6867

68+
from .calibration_dataset_builder import OVCalibrationDatasetBuilder
69+
6970
register_module(ignored_algorithms=[])(Conv1D)
7071

7172
core = Core()
@@ -90,6 +91,11 @@ def __init__(self, model: transformers.PreTrainedModel, task: Optional[str] = No
9091
super().__init__()
9192
self.model = model
9293
self.task = task
94+
95+
if not is_nncf_available():
96+
# TODO: add message
97+
raise Exception
98+
9399
self.dataset_builder = OVCalibrationDatasetBuilder(model, seed)
94100

95101
@classmethod

0 commit comments

Comments
 (0)