"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3323146e904a1092def4e8527de9d2a7479c1c14"
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
......@@ -157,3 +157,6 @@ debug.env
#ctags
tags
# pre-commit
.pre-commit*
......@@ -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
......
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