search

catherinesyeh / attention-viz

Visualizing query-key interactions in language + vision transformers
http://attentionviz.com/
MIT License
122 stars 15 forks source link

Show all queries + keys on click #49

Closed catherinesyeh closed 1 year ago

catherinesyeh commented 1 year ago

Highlight clicked on token, then show remaining queries + keys (instead of currently, it just shows opposite of clicked on token)