Update config.json

config.json

@@ -2,11 +2,13 @@
   "architectures": [
     "LatentRecurrentDepthModel"
   ],
-  "d_model": 768,
-  "dropout": 0.1,
+  "auto_map": {
+    "AutoModelForCausalLM": "modeling_latent_recurrent_depth.LatentRecurrentDepthModel",
+    "AutoConfig": "modeling_latent_recurrent_depth.LatentRecurrentDepthConfig"
+  },
   "model_type": "latent_recurrent_depth",
+  "vocab_size": 50257,
+  "d_model": 768,
   "num_heads": 12,
-  "torch_dtype": "float32",
-  "transformers_version": "4.47.0",
-  "vocab_size": 50257
+  "dropout": 0.1
 }