codebook-features
                                
                                 codebook-features copied to clipboard
                                
                                    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.