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] Revert back the kwargs argument for intervention init #166

Closed frankaging closed 4 days ago

frankaging commented 4 days 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: