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