File size: 573 Bytes
70c8597
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
{
    "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"
}