Requires https://github.com/saprmarks/dictionary_learning/tree/collab.
sae = GatedAutoEncoder.from_pretrained("/path/to/checkpoint.pt")
Requires https://github.com/saprmarks/dictionary_learning/tree/collab.
sae = GatedAutoEncoder.from_pretrained("/path/to/checkpoint.pt")