Downloads · 30 days
5
1% of all-time downloads
ts-large
tdooms/ts-large
ts-large is a machine learning model from tdooms. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
This is the medium version of the bilinear transformers trained on TinyStories. The primary purpose of this model is interpretability, most design choices were made with that in mind.
All-time downloads
903
Public
Parameters
81.8M
327 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors327 MB · 100%
At a glance
- Library
- transformers
- Access
- Public
- Created
- Oct 3, 2024
- Updated
- Nov 19, 2024
- SHA
- 987c69db
Model card
From the Hugging Face model README
TS Large
This is the medium version of the bilinear transformers trained on TinyStories. The primary purpose of this model is interpretability, most design choices were made with that in mind.
The code to run this custom model can be found here, along with many utility functions for weight-based interpretability.
Model Details
- 82 million parameters
- 8 layers
- 12 attention heads
- model dimension 768
- bilinear MLP with expansion factor 4
- context length of 256
- trained for 1 epoch (~500M tokens)
- rotary positional embedding
- custom tinystories tokenizer
- Library
- transformers
- Created
- Oct 3, 2024
- Updated
- Nov 19, 2024