Instructions to use google/gemma-scope-2b-pt-res with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- SAELens
How to use google/gemma-scope-2b-pt-res with SAELens:
# pip install sae-lens from sae_lens import SAE sae, cfg_dict, sparsity = SAE.from_pretrained( release = "RELEASE_ID", # e.g., "gpt2-small-res-jb". See other options in https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml sae_id = "SAE_ID", # e.g., "blocks.8.hook_resid_pre". Won't always be a hook point ) - Notebooks
- Google Colab
- Kaggle
Is there a similar model for CodeGemma 7b IT?
#9
by debadribasak - opened
I want to understand the information captured by layers of codegemma with SAE. Is there a pretrained SAE similar to google/gemma-scope-2b-pt-res for codegemma? If not, can someone guide me on how to train an SAE for code generation and which dataset to use?
Hi we do not have an SAE for that but you can see: https://github.com/decoderesearch/SAELens for general SAE usage and there is a slack link in the README there
ArthurConmyGDM changed discussion status to closed