Beta release CLIP SAE code now part of Prisma Library as v2.0.0!

[soniajoseph]

We now have ViT/CLIP SAE code as part of the Prisma library!

You can see a demo here: https://github.com/soniajoseph/ViT-Prisma/blob/main/demos/Train_CLIP_SAE.ipynb

All the code is in src/vit_prisma/sae.

What's Changed

• Trainer updates for HookedViT by @PraneetNeuro in Trainer updates for HookedViT #66
• Updated usage guide by @PraneetNeuro in Updated usage guide #67
• Conditional hooks and a couple unit tests by @themachinefan in Conditional hooks and a couple unit tests #70
• Base setup for loading pretrained models by @PraneetNeuro in Base setup for loading pretrained models #69
• Loading attn weights fix by @themachinefan in Loading attn weights fix #71
• Folding fix by @themachinefan in Folding fix #73
• ViVit by @themachinefan in ViVit #76
• Hooked prisma by @soniajoseph in Hooked prisma #77
• run tests with github actions by @themachinefan in run tests with github actions #78
• Support loading tiny clip by @themachinefan in Support loading tiny clip #85
• Vision SAEs by @themachinefan in Vision SAEs #90
• upgrade sae-lens to v2.1.3 by @themachinefan in upgrade sae-lens to v2.1.3 #93
• Support for loading huggingface VitForImageClassification by @themachinefan in Support for loading huggingface VitForImageClassification #92
• Added documentation for main demo as listed in Issue Make sure we have documentation for functions in ViT Prisma Main Demo #88 by @jain18ayush in Added documentation for main demo as listed in Issue #88 #100
• Added documentation for sae-train and eval by @jain18ayush in Added documentation for sae-train and eval  #102
• Cleaned up configuration parsing for SAE training by @PraneetNeuro in Cleaned up configuration parsing for SAE training #99
• Sae lens 2 by @soniajoseph in Sae lens 2 #107

New Contributors

• @jain18ayush made their first contribution in Added documentation for main demo as listed in Issue #88 #100

Full Changelog: v0.1.0...v2.0.0

---

This discussion was created from the release Beta release CLIP SAE code now part of Prisma Library as v2.0.0!.