SAEs for use with the SAELens library
This repository contains the following SAEs:
- blocks.0:0.attn.hook_z
- blocks.0:1.attn.hook_z
- blocks.0:2.attn.hook_z
- blocks.0:3.attn.hook_z
- blocks.0:4.attn.hook_z
- blocks.0:5.attn.hook_z
- blocks.0:6.attn.hook_z
- blocks.0:7.attn.hook_z
- blocks.1:0.attn.hook_z
- blocks.1:1.attn.hook_z
- blocks.1:2.attn.hook_z
- blocks.1:3.attn.hook_z
- blocks.1:4.attn.hook_z
- blocks.1:5.attn.hook_z
- blocks.1:6.attn.hook_z
- blocks.1:7.attn.hook_z
Load these SAEs using SAELens as below:
from sae_lens import SAE
sae, cfg_dict, sparsity = SAE.from_pretrained("timaeus/tetrahedron-saes-per-head-attn", "<sae_id>")