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

* [FlaxSpeechEncoderDecoder] Fix bug in `decoder_module` * [FlaxEncoderDecoder] Fix bug in `decoder_module`

[Flax(Speech)EncoderDecoder] Fix bug in `decoder_module` (#17036)
* [FlaxSpeechEncoderDecoder] Fix bug in `decoder_module` * [FlaxEncoderDecoder] Fix bug in `decoder_module`
93b802c4 · Sanchit Gandhi · GitHub · 1ae182d9 · 93b802c4 · 93b802c4
Unverified Commit 93b802c4 authored May 02, 2022 by Sanchit Gandhi Committed by GitHub May 02, 2022
2 changed files
--- a/src/transformers/models/encoder_decoder/modeling_flax_encoder_decoder.py
+++ b/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,
            )


--- a/src/transformers/models/speech_encoder_decoder/modeling_flax_speech_encoder_decoder.py
+++ b/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,
            )