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

sanchit-gandhi · elusenji · commit 3070eaded85e · 2022-06-12T11:14:28.000+02:00
* [FlaxSpeechEncoderDecoder] Fix bug in `decoder_module`

* [FlaxEncoderDecoder] Fix bug in `decoder_module`
diff --git 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 @@ def _decoder_forward(
                 decoder_input_ids,
                 decoder_attention_mask,
                 decoder_position_ids,
-                encoder_hidden_states,
+                encoder_hidden_states=encoder_hidden_states,
                 **kwargs,
             )
 
diff --git 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 @@ def _decoder_forward(
                 decoder_input_ids,
                 decoder_attention_mask,
                 decoder_position_ids,
-                encoder_hidden_states,
+                encoder_hidden_states=encoder_hidden_states,
                 **kwargs,
             )
 

Original file line number	Diff line number	Diff line change
`@@ -593,7 +593,7 @@ def _decoder_forward(`
`593`	`593`	`decoder_input_ids,`
`594`	`594`	`decoder_attention_mask,`
`595`	`595`	`decoder_position_ids,`
`596`		`- encoder_hidden_states,`
	`596`	`+ encoder_hidden_states=encoder_hidden_states,`
`597`	`597`	`**kwargs,`
`598`	`598`	`)`
`599`	`599`
Original file line number	Diff line number	Diff line change
`@@ -627,7 +627,7 @@ def _decoder_forward(`
`627`	`627`	`decoder_input_ids,`
`628`	`628`	`decoder_attention_mask,`
`629`	`629`	`decoder_position_ids,`
`630`		`- encoder_hidden_states,`
	`630`	`+ encoder_hidden_states=encoder_hidden_states,`
`631`	`631`	`**kwargs,`
`632`	`632`	`)`
`633`	`633`