google
/

roberta2roberta_L-24_gigaword

Summarization Transformers PyTorch

English encoder-decoder text2text-generation Inference Endpoints

Model card Files Files and versions Community

system HF staff commited on Sep 10, 2020

Commit

aa98b0e

•

1 Parent(s): 2345333

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
   "decoder": {
     "add_cross_attention": true,
     "architectures": [
-      "BertForSeqGenerationDecoder"
     ],
     "attention_probs_dropout_prob": 0.1,
     "bad_words_ids": null,
@@ -73,7 +73,7 @@
   "encoder": {
     "add_cross_attention": false,
     "architectures": [
-      "BertForSeqGenerationDecoder"
     ],
     "attention_probs_dropout_prob": 0.1,
     "bad_words_ids": null,

   "decoder": {
     "add_cross_attention": true,
     "architectures": [
+      "BertGenerationEncoder"
     ],
     "attention_probs_dropout_prob": 0.1,
     "bad_words_ids": null,
   "encoder": {
     "add_cross_attention": false,
     "architectures": [
+      "BertGenerationDecoder"
     ],
     "attention_probs_dropout_prob": 0.1,
     "bad_words_ids": null,