Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "1933231a0a98ff5710118f3229a865db7625db04"
Unverified Commit 93b802c4 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

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

* [FlaxSpeechEncoderDecoder] Fix bug in `decoder_module`

* [FlaxEncoderDecoder] Fix bug in `decoder_module`
parent 1ae182d9
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/encoder_decoder/modeling_flax_encoder_decoder.py
View file @ 93b802c4
...@@ -593,7 +593,7 @@ class FlaxEncoderDecoderModel(FlaxPreTrainedModel): ...@@ -593,7 +593,7 @@ class FlaxEncoderDecoderModel(FlaxPreTrainedModel):
decoder_input_ids, decoder_input_ids,
decoder_attention_mask, decoder_attention_mask,
decoder_position_ids, decoder_position_ids,
encoder_hidden_states, encoder_hidden_states=encoder_hidden_states,
**kwargs, **kwargs,
) )
......
src/transformers/models/speech_encoder_decoder/modeling_flax_speech_encoder_decoder.py
View file @ 93b802c4
...@@ -627,7 +627,7 @@ class FlaxSpeechEncoderDecoderModel(FlaxPreTrainedModel): ...@@ -627,7 +627,7 @@ class FlaxSpeechEncoderDecoderModel(FlaxPreTrainedModel):
decoder_input_ids, decoder_input_ids,
decoder_attention_mask, decoder_attention_mask,
decoder_position_ids, decoder_position_ids,
encoder_hidden_states, encoder_hidden_states=encoder_hidden_states,
**kwargs, **kwargs,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment