TransformerLensOrg / TransformerLens

A library for mechanistic interpretability of GPT-style language models
https://transformerlensorg.github.io/TransformerLens/
MIT License
1.17k stars 241 forks source link

moved enable hook functionality to separate functions and tested new functions #613

Closed bryce13950 closed 1 month ago

bryce13950 commented 1 month ago

Description

Simplified existing functions, merged duplicate code, and tested new functions to resolve an existing issue in 1.x and make the code easier to both test and understand.

Fixes #611

Type of change

Please delete options that are not relevant.

Screenshots

Please attach before and after screenshots of the change if applicable.

Checklist: