{ | |
"model_config": { | |
"n_layer": 24, | |
"n_head": 16, | |
"n_embd": 1024, | |
"block_size": 1024, | |
"bias": false, | |
"input_vocab_size": 129600, | |
"output_vocab_size": 10048, | |
"dropout": 0.0 | |
}, | |
"model_type": "text", | |
"parameter_count": 446088192, | |
"needs_tokenizer": true, | |
"best_val_loss": 1.2688672196865083, | |
"torch_dtype": "bfloat16" | |
} |