TinyBuddy-30M / config.json
Eeppa's picture
Update config.json
6e74e54 verified
raw
history blame contribute delete
371 Bytes
{
"vocab_size": 50000,
"block_size": 128,
"n_layer": 6,
"n_head": 8,
"n_embd": 256,
"mlp_ratio": 4,
"dropout": 0.0,
"tie_weights": false,
"model_type": "tinybuddy",
"architectures": ["TinyGPT"],
"auto_map": {
"AutoModelForCausalLM": "modeling_tinybuddy.TinyGPT",
"AutoConfig": "modeling_tinybuddy.GPTConfig"
},
"torch_dtype": "float32"
}