From 7d8ecae8ac8298aa8f6f262b38d69c73a5125221 Mon Sep 17 00:00:00 2001 From: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> Date: Tue, 28 May 2024 11:19:21 +0200 Subject: [PATCH] Update optimum/onnxruntime/modeling_decoder.py Co-authored-by: Michael Benayoun --- optimum/onnxruntime/modeling_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py index e3150da05e..2d9be2d757 100644 --- a/optimum/onnxruntime/modeling_decoder.py +++ b/optimum/onnxruntime/modeling_decoder.py @@ -533,7 +533,7 @@ def _from_pretrained( output_dims["logits"][1] = "sequence_length" override_dims = True - # Since https://github.com/huggingface/optimum/pull/871/files + # Since https://github.com/huggingface/optimum/pull/871/ # changed axis notation/naming during export, we need to update the dims for dim in input_dims.keys(): if "past" in dim and input_dims[dim][2] == "past_sequence_length + sequence_length":