[P0] IntervenableModel.save() doesn't save trained model parameters

[aryamanarora]

Contact Details

Come to my office on Gates 3rd floor anytime 👍

What happened?

IntervenableModel.save() doesn't save trained model parameters. This is an issue when you are also training e.g. a classification head on top of our model and want to upload it to HF/save it locally. save_intervention() does this properly though.

Code to produce this issue.

No response