Architecture
Retrieved 2026-09-01 at pinned commit cf967c0a9a04; parameter count from the safetensors index at the same revision.
- Architecture
- gpt_neox
- Layers
- 6
- Hidden size
- 128
- Attention heads
- 4
- Feed-forward width
- 512
- Vocabulary
- 50,304
- Context ceiling
- 2,048
Where the memory goes
Every attention head caches its own keys and values, so the KV cache grows at the full multi-head rate with context.