mrm8488
/

bert-hash-pico-ft-prompt-injection

Text Classification

Model card Files Files and versions

mrm8488 commited on 14 days ago

Commit

a935c3e

·

verified ·

1 Parent(s): 502e8f2

Update config.json

Files changed (1) hide show

config.json +8 -0

config.json CHANGED Viewed

@@ -8,6 +8,14 @@
     "AutoModel": "modeling_bert_hash.BertHashModel",
     "AutoModelForMaskedLM": "modeling_bert_hash.BertHashForMaskedLM",
     "AutoModelForSequenceClassification": "modeling_bert_hash.BertHashForSequenceClassification"
   },
   "classifier_dropout": null,
   "dtype": "float32",

     "AutoModel": "modeling_bert_hash.BertHashModel",
     "AutoModelForMaskedLM": "modeling_bert_hash.BertHashForMaskedLM",
     "AutoModelForSequenceClassification": "modeling_bert_hash.BertHashForSequenceClassification"
+  },
+    "id2label": {
+    "0": "legit",
+    "1": "injection"
+  },
+  "label2id": {
+    "legit": 0,
+    "injection": 1
   },
   "classifier_dropout": null,
   "dtype": "float32",