Remove redundant hidden states

90ab15cb · Lysandre · Lysandre Debut · 9a50828b · 90ab15cb
Commit 90ab15cb authored Jan 08, 2020 by Lysandre Committed by Lysandre Debut Feb 04, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_encoder_decoder.py src/transformers/modeling_encoder_decoder.py +1 -1

No files found.
--- a/src/transformers/modeling_encoder_decoder.py
+++ b/src/transformers/modeling_encoder_decoder.py
@@ -232,7 +232,7 @@ class PreTrainedEncoderDecoder(nn.Module):
            encoder_outputs = ()

        kwargs_decoder["encoder_hidden_states"] = encoder_hidden_states
-        decoder_outputs = self.decoder(decoder_input_ids, encoder_hidden_states, **kwargs_decoder)
+        decoder_outputs = self.decoder(decoder_input_ids, **kwargs_decoder)

        return decoder_outputs + encoder_outputs