Update README.md

README.md

@@ -15,11 +15,11 @@ They are loadable using SAE via a few methods. The preferred method is to use th
 ```python
 import torch
 from transformer_lens import HookedTransformer
-from sae_lens import SparseAutoencoder, ActivationsStore
+from sae_lens import SAE, ActivationsStore
 
 torch.set_grad_enabled(False)
 model = HookedTransformer.from_pretrained("gemma-2b")
-sae, cfg, sparsity  = SparseAutoencoder.from_pretrained(
+sae, cfg, sparsity  = SAE.from_pretrained(
   "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
   "blocks.12.hook_resid_post" # change this to another specific SAE ID in the release if desired. 
 )