storylm-10m / config.json
Compactbot's picture
Add config.json
cafa715 verified
Raw History Blame Contribute Delete
343 Bytes
{
"architectures": ["StoryLMForCausalLM"],
"model_type": "storylm",
"vocab_size": 8192,
"d_model": 256,
"n_heads": 8,
"n_kv_heads": 8,
"n_layers": 8,
"head_dim": 32,
"ffn_dim": 1024,
"max_position_embeddings": 512,
"tie_word_embeddings": true,
"rope_base": 10000.0,
"rms_norm_eps": 1e-05,
"torch_dtype": "float32"
}