Merge pull request #23 from neuralmagic/quantization-refactor-patch

update quantization utils import for latest sparseml refactors
neuralmagic · Mar 15, 2022 · ea79050 · ea79050
2 parents 63b9a73 + 0d93762
commit ea79050
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/export.py b/models/export.py
@@ -18,7 +18,7 @@
 from torch.utils.mobile_optimizer import optimize_for_mobile
 
 from sparseml.pytorch.utils import ModuleExporter
-from sparseml.pytorch.utils.quantization import skip_onnx_input_quantize
+from sparseml.pytorch.sparsification.quantization import skip_onnx_input_quantize
 
 import models
 from models.experimental import attempt_load