ai-safety-foundation / sparse_autoencoder

Sparse Autoencoder for Mechanistic Interpretability
https://ai-safety-foundation.github.io/sparse_autoencoder/
MIT License
171 stars 39 forks source link

Revert "Fix deepspeed for single GPU" #188

Closed alan-cooney closed 7 months ago

alan-cooney commented 7 months ago

Reverts ai-safety-foundation/sparse_autoencoder#187