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

[GenerationOutputs] Fix GenerationOutputs Tests (#9443)

Browse Source 
* fix generation models

* fix led

* fix docs

* add is_decoder

* fix last docstrings

* make style

* fix t5 cross attentions

* correct t5
This commit is contained in:
Patrick von Platen
2021-01-06 19:37:02 +01:00
committed by GitHub
parent 0c96262f7d
commit b8462b5b2a
9 changed files with 89 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/encoder_decoder/modeling_encoder_decoder.py
View File

@@ -455,7 +455,7 @@ class EncoderDecoderModel(PreTrainedModel):
"decoder_attention_mask": decoder_attention_mask,
"decoder_input_ids": decoder_inputs["input_ids"],
"encoder_outputs": encoder_outputs,
"past_key_values": past,
"past_key_values": decoder_inputs["past_key_values"],
"use_cache": use_cache,
}
return input_dict