config.json

{
  "architectures": [
    "GPT"
  ],
  "library_name": "mlx",
  "model_config": {
    "n_embd": 384,
    "n_head": 3,
    "n_kv_head": 3,
    "n_layer": 6,
    "sequence_len": 256,
    "vocab_size": 2075,
    "window_pattern": "SSSL"
  },
  "model_type": "human-chess-mlx",
  "torch_dtype": "bfloat16"
}