katanemo
/

Arch-Guard

Text Classification

Model card Files Files and versions Community

cotran2 commited on 3 days ago

Commit

57842a6

·

verified ·

1 Parent(s): 38395c2

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -9,14 +9,14 @@
   "hidden_size": 768,
   "id2label": {
     "0": "BENIGN",
-    "1": "INJECTION",
     "2": "JAILBREAK"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
     "BENIGN": 0,
-    "INJECTION": 1,
     "JAILBREAK": 2
   },
   "layer_norm_eps": 1e-07,

   "hidden_size": 768,
   "id2label": {
     "0": "BENIGN",
+    "1": "UNHARMFUL",
     "2": "JAILBREAK"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
     "BENIGN": 0,
+    "UNHARMFUL": 1,
     "JAILBREAK": 2
   },
   "layer_norm_eps": 1e-07,