"""
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)