stanfordnlp / pyvene

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

[Minor] Support MistralModel and MistralForCausalLM #135

Closed jiudingsun01 closed 8 months ago

jiudingsun01 commented 8 months ago

Description

adding modelcard and mapping (type_to_module and type_to_dimension) for Mistral and MistralForCausalLM

Testing Done

Tested on my own code; worked perfectly.

Checklist: