v2.4.1

Little update to the code usage, but huge update for memory consumption! TransformerLens now needs almost half the memory it needed previously to boot thanks to a change with how the TransformerLens models are loaded.

What's Changed

• removed einsum causing error when use_atten_result is enabled by @oliveradk in #660
• revised loading to recycle state dict by @bryce13950 in #706

New Contributors

• @oliveradk made their first contribution in #660

Full Changelog: v2.4.0...v2.4.1