Can't load model.
#1
by
Dise - opened
Thanks for sharing the pre-trained model! Currently, I am trying to load it by "model.load_state_dict(torch.load(SAE_MODEL_PATH))", but I got the error: "UnpicklingError: could not find MARK". Can you load it correctly? Thanks.
Hi, sorry I am responding so late. I am currently working on making the model compatible with the SAELens library, which should make it much easier to load. This should be finished within a few days. Sorry about this!
You should now be able to use it with the SAE Lens Library (you'll have to re-download the weights).
tylercosgrove changed discussion status to
closed