jackhhao commited on
Commit
495316a
1 Parent(s): f89d5fa

Update model labels

Browse files
Files changed (2) hide show
  1. config.json +8 -0
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -9,8 +9,16 @@
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
 
 
 
 
12
  "initializer_range": 0.02,
13
  "intermediate_size": 3072,
 
 
 
 
14
  "layer_norm_eps": 1e-12,
15
  "max_position_embeddings": 512,
16
  "model_type": "bert",
 
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "benign",
14
+ "1": "jailbreak"
15
+ },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
+ "label2id": {
19
+ "benign": 0,
20
+ "jailbreak": 1
21
+ },
22
  "layer_norm_eps": 1e-12,
23
  "max_position_embeddings": 512,
24
  "model_type": "bert",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ff3e4db9b74fc25bfff5a979f56fee568b47d114d2fb74e41c150afd46af779
3
  size 438001134
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:450171758333f23800a7283e74f638d11f38a5b56420445ee2d0d2a6e35ee37e
3
  size 438001134