| { |
| "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", |
| "loops": 1, |
| "add_metadata_tokens": true, |
| "metadata_dropout_p": 0.3, |
| "input_size": 4171, |
| "output_size": 4135 |
| }, |
| "tokenizers": { |
| "input_tokenizer": "RealizableMoveTokenizer", |
| "output_tokenizer": "RealizableMoveTokenizer" |
| }, |
| "artifacts": { |
| "pytorch": { |
| "path": "model.safetensors", |
| "format": "safetensors" |
| }, |
| "onnx": { |
| "path": "model.onnx", |
| "interface": "bin_logits_v2", |
| "input_name": "bin_moves", |
| "output_name": "bin_logits", |
| "precision": "fp32" |
| }, |
| "onnx_bf16": { |
| "path": "model-bf16-fp32compute.onnx", |
| "interface": "bin_logits_v2", |
| "input_name": "bin_moves", |
| "output_name": "bin_logits", |
| "precision": "bf16" |
| }, |
| "onnx_int8": { |
| "path": "model-int8-blk128.onnx", |
| "interface": "bin_logits_v2", |
| "input_name": "bin_moves", |
| "output_name": "bin_logits", |
| "precision": "int8" |
| }, |
| "onnx_int8_last": { |
| "path": "model-int8-blk128-last.onnx", |
| "interface": "bin_logits_v1", |
| "input_name": "bin_moves", |
| "output_name": "bin_logits", |
| "precision": "int8" |
| } |
| }, |
| "source": { |
| "run_id": "chess-autocomplete-v1-91m", |
| "checkpoint": "checkpoint-000599999.pt", |
| "checkpoint_iter": 599999 |
| } |
| } |
|
|
