Skip to content

GitLab

Commit 90ab15cb authored by Lysandre's avatar Lysandre Committed by Lysandre Debut
Browse files

Remove redundant hidden states

parent 9a50828b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_encoder_decoder.py
View file @ 90ab15cb
...@@ -232,7 +232,7 @@ class PreTrainedEncoderDecoder(nn.Module): ...@@ -232,7 +232,7 @@ class PreTrainedEncoderDecoder(nn.Module):
encoder_outputs = () encoder_outputs = ()
kwargs_decoder["encoder_hidden_states"] = encoder_hidden_states 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 return decoder_outputs + encoder_outputs
......
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