Update README.md (#1)
Browse files- Update README.md (a22116e00e5f600568f45b588ebf5d9eb8b84e4e)
Co-authored-by: louka ewington-pitsos <[email protected]>
README.md
CHANGED
|
@@ -18,7 +18,7 @@ from transformer_lens import HookedTransformer
|
|
| 18 |
from sae_lens import SAE, ActivationsStore
|
| 19 |
|
| 20 |
torch.set_grad_enabled(False)
|
| 21 |
-
model = HookedTransformer.from_pretrained("gemma-2b")
|
| 22 |
sae, cfg, sparsity = SAE.from_pretrained(
|
| 23 |
"gemma-2b-it-res-jb", # to see the list of available releases, go to: https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml
|
| 24 |
"blocks.12.hook_resid_post" # change this to another specific SAE ID in the release if desired.
|
|
|
|
| 18 |
from sae_lens import SAE, ActivationsStore
|
| 19 |
|
| 20 |
torch.set_grad_enabled(False)
|
| 21 |
+
model = HookedTransformer.from_pretrained("gemma-2b-it")
|
| 22 |
sae, cfg, sparsity = SAE.from_pretrained(
|
| 23 |
"gemma-2b-it-res-jb", # to see the list of available releases, go to: https://github.com/jbloomAus/SAELens/blob/main/sae_lens/pretrained_saes.yaml
|
| 24 |
"blocks.12.hook_resid_post" # change this to another specific SAE ID in the release if desired.
|