[Flax(Speech)EncoderDecoder] Fix bug in `decoder_module` (#17036)

* [FlaxSpeechEncoderDecoder] Fix bug in `decoder_module`

* [FlaxEncoderDecoder] Fix bug in `decoder_module`

src/transformers/models/encoder_decoder/modeling_flax_encoder_decoder.py

@@ -593,7 +593,7 @@ class FlaxEncoderDecoderModel(FlaxPreTrainedModel):
                 decoder_input_ids,
                 decoder_attention_mask,
                 decoder_position_ids,
-                encoder_hidden_states,
+                encoder_hidden_states=encoder_hidden_states,
                 **kwargs,
             )
 

src/transformers/models/speech_encoder_decoder/modeling_flax_speech_encoder_decoder.py

@@ -627,7 +627,7 @@ class FlaxSpeechEncoderDecoderModel(FlaxPreTrainedModel):
                 decoder_input_ids,
                 decoder_attention_mask,
                 decoder_position_ids,
-                encoder_hidden_states,
+                encoder_hidden_states=encoder_hidden_states,
                 **kwargs,
             )