---
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.