diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py index e3150da05e1..2d9be2d757f 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":