Push model using huggingface_hub.

README.md

@@ -1,12 +1,9 @@
 ---
-language: en
-library_name: mlsae
-license: mit
 tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
 ---
 
 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/tim-lawson/mlsae
+- Library: [More Information Needed]
 - Docs: [More Information Needed]

config.json

@@ -1,9 +1,20 @@
 {
+  "accumulate_grad_batches": 64,
   "auxk": 256,
-  "dead_steps_threshold": 76,
+  "auxk_coef": 0.03125,
+  "batch_size": 1,
+  "dead_steps_threshold": null,
   "dead_threshold": 0.001,
+  "dead_tokens_threshold": 10000000,
+  "expansion_factor": 64,
   "k": 32,
-  "n_inputs": 1024,
-  "n_latents": 65536,
-  "standardize": true
+  "layers": [
+    2
+  ],
+  "lr": 0.0001,
+  "max_length": 2048,
+  "model_name": "EleutherAI/pythia-410m-deduped",
+  "skip_special_tokens": true,
+  "standardize": true,
+  "tuned_lens": false
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1b47afc76b3c18570247aef1ab121b5d571d58ba57fdbc6af138a689e787fb7e
-size 536875280
+oid sha256:be8ba3a7f2a6298965343585707aad4b299d7edff656e36eb0dea5f108cf096c
+size 2158251016