search

soniajoseph / ViT-Prisma

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

Write hooked_prisma functionality to take pretrained models #64

Closed soniajoseph closed 7 months ago

soniajoseph commented 8 months ago

Includes both timm models and prisma models trained with last version of the repo

PraneetNeuro commented 8 months ago

I've started work on this issue