PromptShield checkpoints

Index of PromptShield fine-tunes on this account.

Sequence classifiers (Flan-T5)

Model	Base	Checkpoint
robbypambudi/prompt-shield-flan-t5-base	google/flan-t5-base	2026-08-26 best (epoch 0)
robbypambudi/prompt-shield-flan-t5-small	google/flan-t5-small	2026-08-19 best (epoch 2)

DeBERTa (earlier)

Model	Task
robbypambudi/prompt-shield-deberta-binary	binary
robbypambudi/prompt-shield-deberta-multilabel	multilabel

Load a classifier with AutoModelForSequenceClassification.from_pretrained(...). Class 1 is INJECTION.