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

[P0] Revert back the kwargs argument for intervention init #166

Closed frankaging closed 4 months ago

frankaging commented 4 months ago

Description

Recent SAE change causes a regression for some existing tests. This change tries to fix the issue by reverting back some code change.

Testing Done

Running existing tests.

Checklist: