We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 395fc04 + d028006 commit 77a1572Copy full SHA for 77a1572
neural_compressor/torch/algorithms/autoround/autoround.py
@@ -41,9 +41,8 @@ def _is_auto_round_available():
41
42
from neural_compressor.common.utils import Statistics
43
from neural_compressor.torch.algorithms import Quantizer
44
-from neural_compressor.torch.utils import get_accelerator, logger
45
-
46
from neural_compressor.torch.algorithms.weight_only.utility import CapturedDataloader, InputCaptureModule
+from neural_compressor.torch.utils import get_accelerator, logger
47
48
49
class AutoRoundQuantizer(Quantizer):
0 commit comments