[SpeechEncoderDecoderModel] Correct Encoder Last Hidden State Output (#16586)

6f9d8dc1 · Sanchit Gandhi · GitHub · dad5ca83 · 6f9d8dc1
Unverified Commit 6f9d8dc1 authored Apr 04, 2022 by Sanchit Gandhi Committed by GitHub Apr 04, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py ...speech_encoder_decoder/modeling_speech_encoder_decoder.py +1 -1

No files found.
--- a/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
+++ b/src/transformers/models/speech_encoder_decoder/modeling_speech_encoder_decoder.py
@@ -572,7 +572,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel):
            decoder_hidden_states=decoder_outputs.hidden_states,
            decoder_attentions=decoder_outputs.attentions,
            cross_attentions=decoder_outputs.cross_attentions,
-            encoder_last_hidden_state=encoder_outputs.last_hidden_state,
+            encoder_last_hidden_state=encoder_hidden_states,
            encoder_hidden_states=encoder_outputs.hidden_states,
            encoder_attentions=encoder_outputs.attentions,
        )