Verm1ion commited on
Commit
ddf8e32
·
verified ·
1 Parent(s): a9a54a3

chore: add id2label mapping and use canonical _name_or_path

Browse files
Files changed (1) hide show
  1. config.json +11 -3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/content/drive/MyDrive/InjectionSentry/models/deberta_v6_3_mof",
3
  "architectures": [
4
  "DebertaV2ForSequenceClassification"
5
  ],
@@ -31,5 +31,13 @@
31
  "torch_dtype": "float32",
32
  "transformers_version": "4.44.2",
33
  "type_vocab_size": 0,
34
- "vocab_size": 128100
35
- }
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "Verm1ion/injection-sentry-deberta",
3
  "architectures": [
4
  "DebertaV2ForSequenceClassification"
5
  ],
 
31
  "torch_dtype": "float32",
32
  "transformers_version": "4.44.2",
33
  "type_vocab_size": 0,
34
+ "vocab_size": 128100,
35
+ "id2label": {
36
+ "0": "SAFE",
37
+ "1": "INJECTION"
38
+ },
39
+ "label2id": {
40
+ "SAFE": 0,
41
+ "INJECTION": 1
42
+ }
43
+ }