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] Adding back GPT2 and other model supports #138

Closed frankaging closed 3 months ago

frankaging commented 3 months ago

Descriptions:

Recent PR accidentally removes the supports of a couple of models. This PR adds them back. See issue raised: https://github.com/stanfordnlp/pyvene/commit/bf09440999ddc3552e937de9bc7574b2d4820a2f#diff-e5c418b4f46cb08393fe2f1c9698b2d20d5e51e746b6d3f092f05d4f9744dc98