diff --git a/optimum/intel/neural_compressor/quantization.py b/optimum/intel/neural_compressor/quantization.py index bb232cdade..d4846adc15 100644 --- a/optimum/intel/neural_compressor/quantization.py +++ b/optimum/intel/neural_compressor/quantization.py @@ -15,12 +15,12 @@ import copy import inspect import logging +import warnings from enum import Enum from itertools import chain from pathlib import Path from typing import Callable, Dict, Optional, Union -import warnings import torch from datasets import Dataset, load_dataset from neural_compressor.adaptor.pytorch import PyTorch_FXAdaptor, _cfg_to_qconfig, _propagate_qconfig