config.json

{
    "architectures": [
        "ChessForCausalLM"
    ],
    "dtype": "float32",
    "model_type": "chess_lm",
    "n_ctx": 256,
    "n_embd": 128,
    "n_head": 4,
    "n_layer": 4,
    "n_positions": 256,
    "transformers_version": "4.57.3",
    "vocab_size": 313,
    "auto_map": {
        "AutoConfig": "model.ChessConfig",
        "AutoModelForCausalLM": "model.ChessForCausalLM"
    }
}