search

soniajoseph / ViT-Prisma

ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).
Other
165 stars 18 forks source link

Edit plot_javascript function so that it displays the correct image if you switch attention heads #52

Closed soniajoseph closed 8 months ago

soniajoseph commented 8 months ago

Screen Shot 2024-01-13 at 8 34 55 PM Screen Shot 2024-01-13 at 8 35 12 PM

Image between 1 and 2 remains the same even though the datapoints are different

soniajoseph commented 8 months ago

Tagging you @themachinefan but zero pressure! It was already great work :)

themachinefan commented 8 months ago

No problem, I should be able to do it

soniajoseph commented 8 months ago

Looks great and works on attn head notebook