roneneldan/TinyStories
Viewer • Updated • 2.14M • 153k • 1.23k
How to use abmakkeh/transcoder-TinyStories-3M-2x with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("abmakkeh/transcoder-TinyStories-3M-2x", device_map="auto")This is a set of transcoders trained on the TinyStories-3M using the 2.2M sample of the TinyStories dataset, which comes out to roughly 2.2B tokens using the full GPT-Neo tokenizer. We trained the transcoders on the MLP layers 2, 4, and 7 with a dictionary size twice that of the MLP layer. The transcoders are organized by hookpoint.
With the sae library installed, you can access an SAE like this:
from sae import Sae
sae = Sae.load_from_hub("abmakkeh/transcoder-TinyStories-3M-2x", hookpoint="h.1.mlp")
Base model
roneneldan/TinyStories-3M