andrey / config.json
XBFDI's picture
Update config.json
0e28380 verified
Raw
History Blame Contribute Delete
436 Bytes
{
"vocab_size": 1172,
"d_model": 256,
"n_layer": 6,
"n_head": 4,
"ctx": 128,
"dropout": 0.3,
"architectures": [
"EasyFormerLMHeadModel"
],
"model_type": "transformer",
"auto_map": {
"AutoConfig": "configuration_easyformer.EasyFormerConfig",
"AutoModel": "modeling_easyformer.EasyFormerLMHeadModel",
"AutoModelForCausalLM": "modeling_easyformer.EasyFormerLMHeadModel"
},
"torch_dtype": "float32"
}