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.