bert-small-pii-detection
Token-classification model for PII detection, fine-tuned from prajjwal1/bert-small on gravitee-io/pii-detection-dataset.
Params
30 M
Context
512
Downloads 30d
250 K
Likes
7
Download history
daily snapshots · 55 days
▲ 67 K in the last 30 days (21.1%)
316 K236 K
Aug 22Sep 1Sep 11Sep 20
430 K236 K
Jul 28Aug 15Sep 2Sep 20
Can you run it?
Estimated VRAM at 8K context unless noted. Pick your hardware to see the verdict per quantization.
| File | Quant | Size | Est. VRAM | Verdict on RTX 4090 · 24 GB |
|---|---|---|---|---|
| model.safetensors | f32 | 0.1 GB | 0.6 GB | ✅ Runs comfortably |
Estimate: file size × 1.1 + KV cache at 8K + 0.5 GB overhead. Not a benchmark — how we calculate this.
Specifications
- Architecture
- BertForTokenClassification
- Parameters
- 30 M
- Tensor type
- F32
- Context length
- 512
- Vocabulary
- 30,522
- Layers / heads
- 4 / 8
- Licence
- apache-2.0
- First seen on the Hub
- 2025-09-17
- Training datasets
- gravitee-io/pii-detection-dataset
- Added to our catalog
- 2026-07-28
Compare with any token-classification model