Skip to content

Commit 42c8902

Browse files
Apply suggestions from code review
Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>
1 parent 26e08d1 commit 42c8902

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optimum/exporters/openvino/model_configs.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2640,7 +2640,7 @@ def patch_model_for_export(
26402640
def inputs(self):
26412641
common_inputs = super().inputs
26422642
if getattr(self, "stateful", False) and self._behavior == ConfigBehavior.DECODER:
2643-
common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "seq_length"}
2643+
common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "decoder_sequence_length"}
26442644
return common_inputs
26452645

26462646

@@ -2661,7 +2661,7 @@ def patch_model_for_export(
26612661
def inputs(self):
26622662
common_inputs = super().inputs
26632663
if getattr(self, "stateful", False) and self._behavior == ConfigBehavior.DECODER:
2664-
common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "seq_length"}
2664+
common_inputs["decoder_input_ids"] = {0: "batch_size", 1: "decoder_sequence_length"}
26652665
return common_inputs
26662666

26672667

0 commit comments

Comments
 (0)