From 2463448d451ad26d2bb39bcb4125c2260c1b3b95 Mon Sep 17 00:00:00 2001 From: Jambay Kinley Date: Mon, 16 Oct 2023 07:12:26 +0000 Subject: [PATCH] fix import --- olive/passes/onnx/vitis_ai_quantization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/olive/passes/onnx/vitis_ai_quantization.py b/olive/passes/onnx/vitis_ai_quantization.py index 7bf5adf288..afbc95e86f 100644 --- a/olive/passes/onnx/vitis_ai_quantization.py +++ b/olive/passes/onnx/vitis_ai_quantization.py @@ -359,7 +359,7 @@ def _run_for_config( return model_proto_to_olive_model(onnx_model, output_model_path, config) def _quant_preprocess(self, model: ONNXModel, output_model_path: str) -> ONNXModel: - from olive.passes.onnx.quant_pre_process import quant_pre_process + from onnxruntime.quantization.preprocess import quant_pre_process try: quant_pre_process(input_model_path=model.model_path, output_model_path=output_model_path, auto_merge=True)