tinyindianlm-ml / config.json
AdvaithMagic's picture
Export TinyIndianLM checkpoint in HF format
a15230d verified
Raw History Blame Contribute Delete
601 Bytes
{
"model_type": "tinyindianlm",
"architectures": [
"TinyIndianLMForCausalLM"
],
"auto_map": {
"AutoConfig": "configuration_tinyindianlm.TinyIndianLMConfig",
"AutoModelForCausalLM": "modeling_tinyindianlm.TinyIndianLMForCausalLM"
},
"vocab_size": 4192,
"d_model": 512,
"n_layers": 4,
"n_heads": 8,
"n_kv_heads": 4,
"max_seq_len": 768,
"d_ffn": 1408,
"dropout": 0.0,
"rope_theta": 10000.0,
"pad_token_id": 0,
"unk_token_id": 1,
"bos_token_id": 2,
"eos_token_id": 3,
"torch_dtype": "float32",
"transformers_version": "4.0.0"
}