Skip to content

GitLab

Unverified Commit 6f9d8dc1 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

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

parent dad5ca83
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
View file @ 6f9d8dc1
...@@ -572,7 +572,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel): ...@@ -572,7 +572,7 @@ class SpeechEncoderDecoderModel(PreTrainedModel):
decoder_hidden_states=decoder_outputs.hidden_states, decoder_hidden_states=decoder_outputs.hidden_states,
decoder_attentions=decoder_outputs.attentions, decoder_attentions=decoder_outputs.attentions,
cross_attentions=decoder_outputs.cross_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_hidden_states=encoder_outputs.hidden_states,
encoder_attentions=encoder_outputs.attentions, encoder_attentions=encoder_outputs.attentions,
) )
......
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