rlucatoor commited on
Commit
53920ae
·
1 Parent(s): 1408487

update config.json

Browse files
Files changed (1) hide show
  1. config.json +10 -1
config.json CHANGED
@@ -21,5 +21,14 @@
21
  "sinusoidal_pos_embds": false,
22
  "tie_weights_": true,
23
  "transformers_version": "4.57.1",
24
- "vocab_size": 119547
 
 
 
 
 
 
 
 
 
25
  }
 
21
  "sinusoidal_pos_embds": false,
22
  "tie_weights_": true,
23
  "transformers_version": "4.57.1",
24
+ "vocab_size": 119547,
25
+ "id2label": {
26
+ "0": "safe",
27
+ "1": "unsafe"
28
+ },
29
+ "label2id": {
30
+ "safe": 0,
31
+ "unsafe": 1
32
+ },
33
+ "num_labels": 2
34
  }