File size: 1,166 Bytes
9c220eb 81753e2 9c220eb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | {
"format": "chess_autocomplete_model_repo_v1",
"model": {
"architecture": "ChessTransformer",
"dim": 768,
"num_heads": 12,
"num_blocks": 12,
"max_half_moves": 600,
"input_token_representation": "Discrete",
"norm_type": "layernorm",
"mlp_type": "swiglu",
"add_metadata_tokens": true,
"metadata_dropout_p": 0.3,
"input_size": 4169,
"output_size": 4135
},
"tokenizers": {
"input_tokenizer": "RealizableMoveTokenizer",
"output_tokenizer": "RealizableMoveTokenizer"
},
"artifacts": {
"pytorch": {
"path": "model.safetensors",
"format": "safetensors"
},
"onnx": {
"path": "model.onnx",
"interface": "bin_logits_v1",
"input_name": "bin_moves",
"output_name": "bin_logits",
"precision": "fp32"
},
"onnx_bf16": {
"path": "model-bf16.onnx",
"interface": "bin_logits_v1",
"input_name": "bin_moves",
"output_name": "bin_logits",
"precision": "bf16"
}
},
"source": {
"run_id": "chess-transformer_2026-05-19T22-51-30.577771",
"checkpoint": "checkpoint-000599999.pt",
"checkpoint_iter": 599999
}
}
|