testsavantai
/

prompt-injection-defender-base-v0-onnx

Text Classification

prompt-injection-defender

jailbreak-defender

Inference Endpoints

Model card Files Files and versions Community

leobitz commited on Nov 28, 2024

Commit

6e0d942

·

verified ·

1 Parent(s): 3ae4aec

Upload folder using huggingface_hub

Files changed (1) hide show

config.json +3 -4

config.json CHANGED Viewed

@@ -2,13 +2,12 @@
   "_name_or_path": "distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
-    "TsgDistilBertForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "backbone_output_dropout": 0.5,
   "chunk_overlap_size": 64,
   "classifier_dropout": 0.5,
-  "classifier_num_layers": 1,
   "classifier_prob": 0.5,
   "dim": 768,
   "dropout": 0.1,
@@ -25,8 +24,8 @@
   "long_input": false,
   "max_len": 512,
   "max_position_embeddings": 512,
-  "model_name": "tsg-distilbert",
-  "model_type": "tsg-distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,

   "_name_or_path": "distilbert-base-uncased",
   "activation": "gelu",
   "architectures": [
+    "DistilBertForSequenceClassification"
   ],
   "attention_dropout": 0.1,
   "backbone_output_dropout": 0.5,
   "chunk_overlap_size": 64,
   "classifier_dropout": 0.5,
   "classifier_prob": 0.5,
   "dim": 768,
   "dropout": 0.1,
   "long_input": false,
   "max_len": 512,
   "max_position_embeddings": 512,
+  "model_name": "distilbert-base-uncased",
+  "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,