Post
Just put the latest version here: gist.github.com/samwho/6e351... It's messy and vibey and I've only tested it with 2 models so it's possible that different attention schemes confuse it. Looking at specific attention heads near the start of the model (top of grid) is most interesting ime.
0 reposts2 likes