Update README.md

README.md

@@ -3,10 +3,23 @@ language: en
 library_name: mlsae
 license: mit
 tags:
-- model_hub_mixin
-- pytorch_model_hub_mixin
+  - model_hub_mixin
+  - pytorch_model_hub_mixin
+datasets:
+  - monology/pile-uncopyrighted
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: https://github.com/tslwn/mlsae
-- Docs: [More Information Needed]
+# mlsae-pythia-160m-deduped-x8-k32-tfm
+
+A Multi-Layer Sparse Autoencoder (MLSAE) trained on the residual stream
+activation vectors from every layer of
+[EleutherAI/pythia-160m-deduped](https://huggingface.co/EleutherAI/pythia-160m-deduped)
+with an expansion factor of 8 and k = 32, over 1 billion tokens from
+[monology/pile-uncopyrighted](https://huggingface.co/datasets/monology/pile-uncopyrighted).
+This model includes the underlying transformer.
+
+For more details, see:
+
+- Paper: <https://arxiv.org/abs/2409.04185>
+- GitHub repository: <https://github.com/tim-lawson/mlsae>
+- Weights & Biases project: <https://wandb.ai/timlawson-/mlsae>