codebook-features
codebook-features copied to clipboard
Unfreeze transformer_lens version
The rotary embedding issue is transformer_lens breaks the TinyStories codebook models. The issue was introduced in version 1.8.0 and so we have frozen transformer_lens' version at 1.7.0. Bump the version and update the mod_model_classes.py
file once the issue gets resolved.