From ac36596fb81ef68411f35de954588b3850b0d4d8 Mon Sep 17 00:00:00 2001 From: Ye Wang <52801275+wangyems@users.noreply.github.com> Date: Tue, 19 Jan 2021 13:56:34 -0800 Subject: [PATCH] fix convert_common version retrival (#6382) --- onnxruntime/python/tools/transformers/onnx_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/python/tools/transformers/onnx_model.py b/onnxruntime/python/tools/transformers/onnx_model.py index 18015f54fc..02a953e713 100644 --- a/onnxruntime/python/tools/transformers/onnx_model.py +++ b/onnxruntime/python/tools/transformers/onnx_model.py @@ -408,9 +408,9 @@ class OnnxModel: """ Convert a graph to FLOAT16 """ from packaging.version import Version - import onnxconverter_common.float16 as oc + import onnxconverter_common as oc if Version(oc.__version__) > Version("1.7.0"): - self.model = oc.convert_float_to_float16(self.model, keep_io_types=cast_input_output) + self.model = oc.float16.convert_float_to_float16(self.model, keep_io_types=cast_input_output) return graph = self.model.graph