A. Y. commited on
Commit
53633ef
·
verified ·
1 Parent(s): 22a1269

removed MLM head for API use with feature extraction

Browse files
Files changed (1) hide show
  1. config.json +5 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "BertForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "classifier_dropout": null,
@@ -20,5 +20,8 @@
20
  "transformers_version": "4.41.2",
21
  "type_vocab_size": 2,
22
  "use_cache": true,
23
- "vocab_size": 30522
 
 
 
24
  }
 
1
  {
2
  "architectures": [
3
+ "BertModel"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "classifier_dropout": null,
 
20
  "transformers_version": "4.41.2",
21
  "type_vocab_size": 2,
22
  "use_cache": true,
23
+ "vocab_size": 30522,
24
+ "task_specific_params": {
25
+ "feature-extraction": {}
26
+ }
27
  }