Skip to content

Commit

Permalink
add loader import to readme
Browse files Browse the repository at this point in the history
  • Loading branch information
jbloom-md committed Dec 5, 2023
1 parent 88f086b commit b63f14e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,7 @@ sparse_autoencoder = language_model_sae_runner(cfg)
## Loading a Pretrained Language Model

```python
from sae_training.utils import LMSparseAutoencoderSessionloader

path ="path/to/sparse_autoencoder.pt"
model, sparse_autoencoder, activations_loader = LMSparseAutoencoderSessionloader.load_session_from_pretrained(
Expand Down

0 comments on commit b63f14e

Please sign in to comment.