D0TheMath's picture
Add README.md
91645c6 verified
|
raw
history blame
697 Bytes
metadata
library_name: saelens

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