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

[Minor] Support MistralModel and MistralForCausalLM #135

Closed jiudingsun01 closed 3 months ago

jiudingsun01 commented 3 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: