Update trained_model/config.json

trained_model/config.json

@@ -2,10 +2,15 @@
   "architectures": [
     "RecursiveLanguageModel"
   ],
+  "auto_map": {
+    "AutoConfig": "mixture_of_recursion.RecursiveLanguageModelConfig",
+    "AutoModel": "mixture_of_recursion.RecursiveLanguageModel",
+    "AutoModelForCausalLM": "mixture_of_recursion.RecursiveLanguageModel"
+  },
   "bos_token_id": 50256,
+  "embedding_dim": 512,
   "dropout_rate": 0.1,
   "dtype": "float32",
-  "embedding_dim": 512,
   "eos_token_id": 50256,
   "max_position_embeddings": 256,
   "max_recursion_steps": 2,
@@ -13,6 +18,7 @@
   "num_attention_heads": 8,
   "num_layers": 6,
   "pad_token_id": 50256,
-  "transformers_version": "4.57.3",
+  "tie_word_embeddings": true,
+  "transformers_version": "4.35.0",
   "vocab_size": 50257
-}
+}