File size: 361 Bytes
5f2451e | 1 2 3 4 5 6 7 8 9 10 11 | """
Minimal example: load one of the InterpGPT Phase 5 TopK SAEs from HuggingFace.
"""
from huggingface_hub import snapshot_download
from sae_lens import SAE
repo_id = "connaaa/interpgpt-sae-phase5"
local = snapshot_download(repo_id=repo_id, allow_patterns=["adhd_L2_hook_resid_post/*"])
sae = SAE.load_from_disk(f"{local}/adhd_L2_hook_resid_post")
print(sae)
|