SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Remove redundant hidden states

Browse Source
This commit is contained in:
Lysandre
2020-01-08 13:52:46 +01:00
committed by Lysandre Debut
parent 9a50828b5c
commit 90ab15cb7a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_encoder_decoder.py
View File

@@ -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