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

stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions

http://pyvene.ai

Apache License 2.0

545 stars 46 forks source link

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

Open aryamanarora opened 2 months ago

aryamanarora commented 2 months ago

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