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-60k", "<sae_id>")
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference API
Unable to determine this model’s pipeline type. Check the docs .

Collection including timaeus/tetrahedron-saes-per-head-attn-60k