Instructions to use ritengzhang/SAE-AIU with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- SAELens
How to use ritengzhang/SAE-AIU with SAELens:
# pip install sae-lens from sae_lens import SAE sae, cfg_dict, sparsity = SAE.from_pretrained( release = "RELEASE_ID", # e.g., "gpt2-small-res-jb". See other options in https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml sae_id = "SAE_ID", # e.g., "blocks.8.hook_resid_pre". Won't always be a hook point ) - Notebooks
- Google Colab
- Kaggle
Commit History
13 days ago
-
Update GitHub link after username change
Verified signature03e8562 -
Add arXiv link
Verified signatureeb56c5f
-
Add gpt2 SAE checkpoints
Verified signature283e0cc -
Add opt-125m SAE checkpoints
Verified signatureaacdcb8 -
Add tiny-stories SAE checkpoints
Verified signature739b27f -
Add training configurations
Verified signatureefa1aa9 -
Add MIT license
Verified signaturec079c2d -
Add model card
Verified signature5fcd0e7 -
initial commit
Verified signature610dab4