system HF staff commited on
Commit
35e5fdc
1 Parent(s): 5993d00

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,
@@ -68,7 +68,7 @@
68
  "encoder": {
69
  "add_cross_attention": false,
70
  "architectures": [
71
- "BertForSeqGenerationEncoderModel"
72
  ],
73
  "attention_probs_dropout_prob": 0.1,
74
  "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,
 
68
  "encoder": {
69
  "add_cross_attention": false,
70
  "architectures": [
71
+ "BertGenerationDecoder"
72
  ],
73
  "attention_probs_dropout_prob": 0.1,
74
  "bad_words_ids": null,