TransformerLensOrg / CircuitsVis

Mechanistic Interpretability Visualizations using React
https://alan-cooney.github.io/CircuitsVis/
MIT License
175 stars 28 forks source link

Shrinkage of `attention_heads` plots #61

Closed callummcdougall closed 1 year ago

callummcdougall commented 1 year ago

Weird shrinkage of cv.attention.attention_heads plots.

callummcdougall commented 1 year ago

https://github.com/alan-cooney/CircuitsVis/assets/45238458/ac34d5a5-d21e-43f4-8734-e2b3589cf183

alan-cooney commented 1 year ago

Now fixed