|
--- |
|
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: |
|
```python |
|
from sae_lens import SAE |
|
|
|
sae, cfg_dict, sparsity = SAE.from_pretrained("timaeus/tetrahedron-saes-per-head-attn", "<sae_id>") |
|
``` |