stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions
http://pyvene.ai
Apache License 2.0
608 stars 59 forks source link

[Minor] Expose more APIs for loading and saving #127

Closed frankaging closed 7 months ago

frankaging commented 7 months ago

Description

Exposing more APIs for loading and saving with trainable model components.

Testing Done

Nothing will be effected. Run P0 tests.

Checklist: