search

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

[Bug]: RoBERTa giving error with Pyvene 0.1.2. for ReFT. "forward() got an unexpected keyword argument 'use_cache'". #163

Open m-dev12 opened 1 week ago

m-dev12 commented 1 week ago

Contact Details

dm5927@nyu.edu

What happened?

RoBERTa does not work with ReFT and gives error when Pyvene 0.1.2 is used. "forward() got an unexpected keyword argument 'use_cache'". This should be related to this fix = https://github.com/stanfordnlp/pyvene/pull/152

Code to produce this issue.

No response