From 597e964710bc416bdbbc8d24fba2d9ec70259a8c Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Fri, 4 Mar 2022 13:56:10 -0500 Subject: [PATCH] update quantization utils import for latest sparseml refactors --- models/export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/export.py b/models/export.py index 278fb476e783..6043c84c6246 100644 --- 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