system HF staff commited on
Commit
aa98b0e
1 Parent(s): 2345333

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -6,7 +6,7 @@
6
  "decoder": {
7
  "add_cross_attention": true,
8
  "architectures": [
9
- "BertForSeqGenerationDecoder"
10
  ],
11
  "attention_probs_dropout_prob": 0.1,
12
  "bad_words_ids": null,
@@ -73,7 +73,7 @@
73
  "encoder": {
74
  "add_cross_attention": false,
75
  "architectures": [
76
- "BertForSeqGenerationDecoder"
77
  ],
78
  "attention_probs_dropout_prob": 0.1,
79
  "bad_words_ids": null,
 
6
  "decoder": {
7
  "add_cross_attention": true,
8
  "architectures": [
9
+ "BertGenerationEncoder"
10
  ],
11
  "attention_probs_dropout_prob": 0.1,
12
  "bad_words_ids": null,
 
73
  "encoder": {
74
  "add_cross_attention": false,
75
  "architectures": [
76
+ "BertGenerationDecoder"
77
  ],
78
  "attention_probs_dropout_prob": 0.1,
79
  "bad_words_ids": null,