File tree 1 file changed +2
-2
lines changed
optimum/exporters/openvino
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -2640,7 +2640,7 @@ def patch_model_for_export(
2640
2640
def inputs (self ):
2641
2641
common_inputs = super ().inputs
2642
2642
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 " }
2644
2644
return common_inputs
2645
2645
2646
2646
@@ -2661,7 +2661,7 @@ def patch_model_for_export(
2661
2661
def inputs (self ):
2662
2662
common_inputs = super ().inputs
2663
2663
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 " }
2665
2665
return common_inputs
2666
2666
2667
2667
You can’t perform that action at this time.
0 commit comments