From 90ab15cb7a8fcf8bf58c05453ddf1aa6a4fa00c1 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Wed, 8 Jan 2020 13:52:46 +0100 Subject: [PATCH] Remove redundant hidden states --- src/transformers/modeling_encoder_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_encoder_decoder.py b/src/transformers/modeling_encoder_decoder.py index 0951baff7d..95d56cf6ac 100644 --- 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