google
/

roberta2roberta_L-24_cnn_daily_mail

encoder-decoder

text2text-generation

Inference Endpoints

Model card Files Files and versions Community

system HF staff commited on Sep 10, 2020

Commit

91a4961

•

1 Parent(s): 026f797

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,
@@ -68,7 +68,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,