SAEs for use with the SAELens library

This repository contains the following SAEs:

  • blocks.0.hook_resid_post
  • blocks.1.hook_resid_post
  • blocks.2.hook_resid_post
  • blocks.3.hook_resid_post
  • blocks.4.hook_resid_post
  • blocks.5.hook_resid_post
  • blocks.6.hook_resid_post
  • blocks.7.hook_resid_post
  • blocks.8.hook_resid_post
  • blocks.9.hook_resid_post
  • blocks.10.hook_resid_post
  • blocks.11.hook_resid_post

Load these SAEs using SAELens as below:

from sae_lens import SAE

sae, cfg_dict, sparsity = SAE.from_pretrained("suchitg/sae-compression-gpt-2-small-pruned-sae-pile_0.5", "<sae_id>")
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support