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
416 K
Likes
6
Download history
daily snapshots · 10 days430 K379 K
Jul 28Jul 31Aug 3Aug 6
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
Sponsored · GPU cloud
Not enough VRAM?
Spin up a 24 GB L4 instance in 40 seconds. $0.44/hr.