update quantization utils import for latest sparseml refactors

neuralmagic · Apr 14, 2022 · 597e964 · 597e964
1 parent 4c24632
commit 597e964
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