diff --git a/onnxruntime/python/tools/transformers/fusion_embedlayer.py b/onnxruntime/python/tools/transformers/fusion_embedlayer.py index ccf6f426f1..25cc133755 100644 --- a/onnxruntime/python/tools/transformers/fusion_embedlayer.py +++ b/onnxruntime/python/tools/transformers/fusion_embedlayer.py @@ -116,7 +116,7 @@ class FusionEmbedLayerNoMask(Fusion): if path1 is None: return False - expand, shape = path1 + expand, shape = path1[0], path1[-1] if shape.input[0] != input_ids: return False