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