We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 089742a commit 9430beaCopy full SHA for 9430bea
optimum/intel/openvino/quantization/calibration_dataset_builder.py
@@ -21,7 +21,6 @@
21
import torch
22
import transformers
23
from huggingface_hub.constants import HUGGINGFACE_HUB_CACHE
24
-from nncf.torch.initialization import PTInitializingDataLoader
25
from PIL import Image
26
from torch.utils.data import DataLoader, RandomSampler
27
from tqdm import tqdm
@@ -50,6 +49,7 @@
50
49
51
if is_nncf_available():
52
import nncf
+ from nncf.torch.initialization import PTInitializingDataLoader
53
54
logger = logging.getLogger(__name__)
55
0 commit comments