Use MPS (Apple Silicon) when available - Githubissues

vgel / repeng

A library for making RepE control vectors

https://vgel.me/posts/representation-engineering/

MIT License

435 stars 31 forks source link

Use MPS (Apple Silicon) when available #8

Closed tonytheodore closed 5 months ago

tonytheodore commented 5 months ago

Thank you! Fascinating project.

M1 with 64gb on the vector_ops notebook: