Chess Challenge submission by Drilly93

README.md

@@ -14,13 +14,13 @@ Chess model submitted to the LLM Course Chess Challenge.
 ## Submission Info
 
 - **Submitted by**: [Drilly93](https://huggingface.co/Drilly93)
-- **Parameters**: 475,584
+- **Parameters**: 937,600
 - **Organization**: LLM-course
 
 ## Model Details
 
 - **Architecture**: Chess Transformer (GPT-style)
 - **Vocab size**: 86
-- **Embedding dim**: 96
+- **Embedding dim**: 128
 - **Layers**: 6
 - **Heads**: 4

config.json

@@ -8,10 +8,10 @@
   "eos_token_id": 2,
   "layer_norm_epsilon": 1e-05,
   "model_type": "chess_transformer",
-  "n_ctx": 192,
-  "n_embd": 96,
+  "n_ctx": 256,
+  "n_embd": 128,
   "n_head": 4,
-  "n_inner": 192,
+  "n_inner": 320,
   "n_layer": 6,
   "pad_token_id": 0,
   "tie_weights": true,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7cef042c2edf862c94ec89087b56736afc19a2bdbbc053b691955ca74698e539
-size 1908680
+oid sha256:68bba7b575d04fdb9970c196c970e35e83dc8aab38b2fa08be483c13088ac904
+size 3756848

tokenizer_config.json

@@ -40,8 +40,5 @@
   "model_max_length": 1000000000000000019884624838656,
   "pad_token": "[PAD]",
   "tokenizer_class": "ChessTokenizer",
-  "auto_map": {
-    "AutoTokenizer": "tokenizer.ChessTokenizer"
-  },
   "unk_token": "[UNK]"
 }