Skip to content

GitLab

Unverified Commit c912ba5f authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[EncoderDecoder] Fix Typo (#7915) 

* fix encoder decoder models

* add .gitignore
parent 55bcd0cb
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
.gitignore
View file @ c912ba5f
...@@ -157,3 +157,6 @@ debug.env ...@@ -157,3 +157,6 @@ debug.env
#ctags #ctags
tags tags
# pre-commit
.pre-commit*
src/transformers/modeling_encoder_decoder.py
View file @ c912ba5f
...@@ -434,8 +434,6 @@ class EncoderDecoderModel(PreTrainedModel): ...@@ -434,8 +434,6 @@ class EncoderDecoderModel(PreTrainedModel):
encoder_attentions=encoder_outputs.attentions, encoder_attentions=encoder_outputs.attentions,
) )
return decoder_outputs + encoder_outputs
def prepare_inputs_for_generation(self, input_ids, past, attention_mask, encoder_outputs, **kwargs): 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_inputs = self.decoder.prepare_inputs_for_generation(input_ids)
decoder_attention_mask = decoder_inputs["attention_mask"] if "attention_mask" in decoder_inputs else None decoder_attention_mask = decoder_inputs["attention_mask"] if "attention_mask" in decoder_inputs else None
......
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