google
/

bert2bert_L-24_wmt_de_en

encoder-decoder

text2text-generation

Model card Files Files and versions

system HF Staff commited on Sep 10, 2020

Commit

35e5fdc

·

1 Parent(s): 5993d00

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": [
-      "BertForSeqGenerationEncoderModel"
     ],
     "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,