Text Classification
Transformers
ONNX
Safetensors
Transformers.js
English
electra
prompt-injection
jailbreak-detection
security
Instructions to use yzdvai/tiny-injection-guard with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use yzdvai/tiny-injection-guard with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="yzdvai/tiny-injection-guard")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("yzdvai/tiny-injection-guard") model = AutoModelForSequenceClassification.from_pretrained("yzdvai/tiny-injection-guard", device_map="auto") - Transformers.js
How to use yzdvai/tiny-injection-guard with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('text-classification', 'yzdvai/tiny-injection-guard'); - Notebooks
- Google Colab
- Kaggle
Add ONNX export (fp32 and dynamic int8) for transformers.js
Browse files- onnx/model.onnx +3 -0
- onnx/model_quantized.onnx +3 -0
onnx/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1de0b11bf9fe794380df48661b62962eea49caff36d1f934b195364bc0cef7f4
|
| 3 |
+
size 54420416
|
onnx/model_quantized.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:23fa2692a8278f180178cfa71e1388daa22bcb90f32921f55e77cb111e02c82c
|
| 3 |
+
size 14084770
|