Apply suggestions from code review

Co-authored-by: Ilyas Moutawwakil <[email protected]>
huggingface · Jan 13, 2025 · 42c8902 · 42c8902
1 parent 26e08d1
commit 42c8902
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py
@@ -2640,7 +2640,7 @@ def patch_model_for_export(
     def inputs(self):
         common_inputs = super().inputs
         if getattr(self, "stateful", False) and self._behavior == ConfigBehavior.DECODER:
-            common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "seq_length"}
+            common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "decoder_sequence_length"}
         return common_inputs
 
 
@@ -2661,7 +2661,7 @@ def patch_model_for_export(
     def inputs(self):
         common_inputs = super().inputs
         if getattr(self, "stateful", False) and self._behavior == ConfigBehavior.DECODER:
-            common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "seq_length"}
+            common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "decoder_sequence_length"}
         return common_inputs