From 42c8902c47725de55e8e971e36cd89731e5b489e Mon Sep 17 00:00:00 2001 From: Ekaterina Aidova Date: Mon, 13 Jan 2025 09:05:28 +0400 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> --- optimum/exporters/openvino/model_configs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py index 0d1237b1c..4b1dbb50b 100644 --- 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