Prompt Guard - Prompt Injection & Jailbreak Detection

Prompt Guard is an 86M parameter classifier that detects prompt injections and jailbreak attempts in LLM applications.

Quick start

Installation:

pip install transformers torch

Basic usage:

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
from torch.nn.functional import softmax