File size: 733 Bytes
656c7cc | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | ---
language: en
tags: [text-correction, spelling, punctuation, from-scratch]
---
# WriteFix models
Two text-correction models built and trained from scratch (no pretrained weights). They fix spelling, punctuation, capitals and common grammar slips in English.
| Model | File | Parameters | Size |
|---|---|---|---|
| WF-1 Mini | `wf-1-mini.pt` | 10.5M | 21 MB |
| WF-1 Large | `wf-1-large.pt` | 93.6M | 187 MB |
Both are a conv stem plus a Transformer encoder-decoder with a 16k byte-level BPE vocabulary, stored as half-precision PyTorch state dicts with their config. They load with the WriteFix code, not with `transformers`.
Website and live demo: https://rishiraj38.github.io/writefix-website/
Prototype. Expect mistakes.
|