config.json

{
  "activation_checkpointing": false,
  "architectures": [
    "ChessTransitionPolicy"
  ],
  "auto_map": {
    "AutoConfig": "configuration_chess_policy.ChessPolicyConfig",
    "AutoModel": "modeling_chess_policy.ChessTransitionPolicy"
  },
  "board_layers": 6,
  "board_squares": 64,
  "candidate_layers": 12,
  "d_model": 384,
  "dropout": 0.0,
  "dtype": "float32",
  "model_type": "chess_transition_policy",
  "n_heads": 6,
  "piece_states": 13,
  "swiglu_hidden": 1040,
  "transformers_version": "5.5.3"
}