kaushik-harsh-99's picture
Upload exported SimpleStories Hindi 10M/11M model with safetensors, custom modeling.py and tokenizer
70c8597 verified
Raw
History Blame Contribute Delete
573 Bytes
{
"architectures": [
"SimpleStoriesForCausalLM"
],
"model_type": "simple_stories",
"auto_map": {
"AutoConfig": "modeling_simple_stories.SimpleStoriesConfig",
"AutoModelForCausalLM": "modeling_simple_stories.SimpleStoriesForCausalLM"
},
"vocab_size": 4000,
"max_seq_len": 512,
"d_model": 320,
"n_layers": 7,
"n_heads": 5,
"multiple_of": 64,
"norm_eps": 1e-05,
"dropout": 0.0,
"bos_token_id": 2,
"eos_token_id": 3,
"pad_token_id": 0,
"unk_token_id": 1,
"torch_dtype": "float32"
}