saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[EncoderDecoder] Fix Typo (#7915)

Browse source 
* fix encoder decoder models

* add .gitignore
This commit is contained in:
Patrick von Platen 2020-10-19 22:02:42 +02:00 committed by GitHub
parent 55bcd0cb59
commit c912ba5f69
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.gitignore vendored
View file

@ -157,3 +157,6 @@ debug.env
#ctags
tags
# pre-commit
.pre-commit*

2
src/transformers/modeling_encoder_decoder.py
View file

@ -434,8 +434,6 @@ class EncoderDecoderModel(PreTrainedModel):
encoder_attentions=encoder_outputs.attentions,
)
return decoder_outputs + encoder_outputs
def prepare_inputs_for_generation(self, input_ids, past, attention_mask, encoder_outputs, **kwargs):
decoder_inputs = self.decoder.prepare_inputs_for_generation(input_ids)
decoder_attention_mask = decoder_inputs["attention_mask"] if "attention_mask" in decoder_inputs else None