diff --git a/.gitignore b/.gitignore index cb6316a69..b5d8fc1ec 100644 --- a/.gitignore +++ b/.gitignore @@ -157,3 +157,6 @@ debug.env #ctags tags + +# pre-commit +.pre-commit* diff --git a/src/transformers/modeling_encoder_decoder.py b/src/transformers/modeling_encoder_decoder.py index 9ce246748..ea4044d10 100644 --- a/src/transformers/modeling_encoder_decoder.py +++ b/src/transformers/modeling_encoder_decoder.py @@ -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