system HF staff commited on
Commit
ffe2c37
1 Parent(s): dbaff87

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -8
config.json CHANGED
@@ -1,27 +1,22 @@
1
  {
 
2
  "architectures": [
3
  "DistilBertForMaskedLM"
4
  ],
5
- "activation": "gelu",
6
  "attention_dropout": 0.1,
7
  "dim": 768,
8
  "dropout": 0.1,
9
- "finetuning_task": null,
10
  "hidden_dim": 3072,
11
  "initializer_range": 0.02,
12
  "max_position_embeddings": 512,
 
13
  "n_heads": 12,
14
  "n_layers": 6,
15
- "num_labels": 2,
16
- "output_attentions": false,
17
- "output_hidden_states": false,
18
  "output_past": true,
19
- "pruned_heads": {},
20
  "qa_dropout": 0.1,
21
  "seq_classif_dropout": 0.2,
22
  "sinusoidal_pos_embds": true,
23
  "tie_weights_": true,
24
- "torchscript": false,
25
- "use_bfloat16": false,
26
  "vocab_size": 31102
27
  }
 
1
  {
2
+ "activation": "gelu",
3
  "architectures": [
4
  "DistilBertForMaskedLM"
5
  ],
 
6
  "attention_dropout": 0.1,
7
  "dim": 768,
8
  "dropout": 0.1,
 
9
  "hidden_dim": 3072,
10
  "initializer_range": 0.02,
11
  "max_position_embeddings": 512,
12
+ "model_type": "distilbert",
13
  "n_heads": 12,
14
  "n_layers": 6,
 
 
 
15
  "output_past": true,
16
+ "pad_token_id": 0,
17
  "qa_dropout": 0.1,
18
  "seq_classif_dropout": 0.2,
19
  "sinusoidal_pos_embds": true,
20
  "tie_weights_": true,
 
 
21
  "vocab_size": 31102
22
  }