Update optimum/onnxruntime/modeling_decoder.py

Co-authored-by: Michael Benayoun <[email protected]>
huggingface · May 28, 2024 · 7d8ecae · 7d8ecae
1 parent e4d259e
commit 7d8ecae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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":