Update config.json

config.json

@@ -1,20 +1,22 @@
 {
-  "architectures": [
-    "ChessForCausalLM"
-  ],
-  "bos_token_id": 1,
-  "dropout": 0.1,
-  "dtype": "float32",
-  "eos_token_id": 2,
-  "layer_norm_epsilon": 1e-05,
+  "architectures": ["ChessForCausalLM"],
+  "auto_map": {
+    "AutoConfig": "model.ChessConfig",
+    "AutoModelForCausalLM": "model.ChessForCausalLM"
+  },
   "model_type": "chess_transformer",
+  "vocab_size": 85,
   "n_ctx": 512,
   "n_embd": 100,
   "n_head": 5,
   "n_inner": 384,
   "n_layer": 8,
+  "dropout": 0.1,
+  "layer_norm_epsilon": 1e-05,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
   "pad_token_id": 0,
   "tie_weights": true,
-  "transformers_version": "4.57.3",
-  "vocab_size": 85
+  "torch_dtype": "float32",
+  "transformers_version": "4.57.3"
 }