Chess Challenge submission by Tr1ckst3rrr

README.md

@@ -14,13 +14,13 @@ Chess model submitted to the LLM Course Chess Challenge.
 ## Submission Info
 
 - **Submitted by**: [Tr1ckst3rrr](https://huggingface.co/Tr1ckst3rrr)
-- **Parameters**: 909,824
+- **Parameters**: 975,744
 - **Organization**: LLM-course
 
 ## Model Details
 
 - **Architecture**: Chess Transformer (GPT-style)
-- **Vocab size**: 1682
-- **Embedding dim**: 128
-- **Layers**: 4
-- **Heads**: 4
+- **Vocab size**: 4074
+- **Embedding dim**: 96
+- **Layers**: 6
+- **Heads**: 6

config.json

@@ -9,12 +9,12 @@
   "layer_norm_epsilon": 1e-05,
   "model_type": "chess_transformer",
   "n_ctx": 256,
-  "n_embd": 128,
-  "n_head": 4,
-  "n_inner": 384,
-  "n_layer": 4,
+  "n_embd": 96,
+  "n_head": 6,
+  "n_inner": 288,
+  "n_layer": 6,
   "pad_token_id": 0,
   "tie_weights": true,
   "transformers_version": "4.57.5",
-  "vocab_size": 1682
+  "vocab_size": 4074
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a09dfd42913185e2f31eb31196700ece36e4df2cac7ba9fa0d970924754696cf
-size 3643696
+oid sha256:c75c506fe50867cbefd146f20038da58019682bf2cacdb217ea8cdfee51c61ef
+size 3909328