jbloomAus / SAELens

Training Sparse Autoencoders on Language Models
https://jbloomaus.github.io/SAELens/
MIT License
193 stars 67 forks source link

[Bug Report] SAE.from_pretrained errors out in Hooked_SAE_Transformer_Demo.ipynb #193

Closed babcockt18 closed 6 days ago

babcockt18 commented 1 week ago

Describe the bug SAE.from_pretrained in tutorials/Hooked_SAE_Transformer_Demo.ipynb returns TypeError: SAEConfig.__init__() missing 1 required positional argument: 'dataset_trust_remote_code'

Code example I ran the cells in the proper order and returned the below errors image

System Info

Additional context My current hunch is that the addition of dataset_trust_remote_code wasn't propogated to the preconfigured dictionary which needs the parameter added to fix the issue.

Checklist

aayushk61466 commented 6 days ago

hey, I am also facing the same issue. @babcockt18 is there a way around this?

jbloomAus commented 6 days ago

should be fixed