Instructions to use callummcdougall/arena-demos-transcoder with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- SAELens
How to use callummcdougall/arena-demos-transcoder 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
Update gpt2-small-layer-8-mlp-transcoder/cfg.json
Browse files
gpt2-small-layer-8-mlp-transcoder/cfg.json
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
{"architecture": "standard", "d_in": 768, "d_sae": 24576, "dtype": "float32", "device": "cuda", "model_name": "gpt2-small", "hook_name": "blocks.8.ln2.hook_normalized", "hook_layer": 8, "hook_head_index": null, "activation_fn_str": "relu", "activation_fn_kwargs": {}, "apply_b_dec_to_input": true, "finetuning_scaling_factor": false, "sae_lens_training_version": null, "prepend_bos": true, "dataset_path": "Skylion007/openwebtext", "dataset_trust_remote_code": true, "context_size": 128, "normalize_activations": false, "neuronpedia_id": null, "model_from_pretrained_kwargs": {}}
|
|
|
|
| 1 |
+
{"architecture": "standard", "d_in": 768, "d_sae": 24576, "dtype": "float32", "device": "cuda", "model_name": "gpt2-small", "hook_name": "blocks.8.ln2.hook_normalized", "hook_layer": 8, "hook_head_index": null, "activation_fn_str": "relu", "activation_fn_kwargs": {}, "apply_b_dec_to_input": true, "finetuning_scaling_factor": false, "sae_lens_training_version": null, "prepend_bos": true, "dataset_path": "Skylion007/openwebtext", "dataset_trust_remote_code": true, "context_size": 128, "normalize_activations": false, "use_b_dec_out": true, "neuronpedia_id": null, "model_from_pretrained_kwargs": {}}
|