From 512c803550e05682d02eae2e34efadc5fccfed79 Mon Sep 17 00:00:00 2001 From: Xiaoyu <85524621+xiaoyu-work@users.noreply.github.com> Date: Tue, 26 Mar 2024 20:16:09 -0700 Subject: [PATCH] fix import in transformer optimizer python script (#20091) ### Description Fix import. ### Motivation and Context Fix error. --- onnxruntime/python/tools/transformers/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/transformers/optimizer.py b/onnxruntime/python/tools/transformers/optimizer.py index 068ccefef7..e334e22d92 100644 --- a/onnxruntime/python/tools/transformers/optimizer.py +++ b/onnxruntime/python/tools/transformers/optimizer.py @@ -40,9 +40,9 @@ from onnx_model_t5 import T5OnnxModel from onnx_model_tnlr import TnlrOnnxModel from onnx_model_unet import UnetOnnxModel from onnx_model_vae import VaeOnnxModel +from onnx_utils import extract_raw_data_from_model, has_external_data import onnxruntime -from onnxruntime.transformers.onnx_utils import extract_raw_data_from_model, has_external_data logger = logging.getLogger(__name__)