Commit
65a1ed5
·
verified ·
1 Parent(s): 493005d

Upload speedai-125M-config.json

Browse files
Files changed (1) hide show
  1. speedai-125M-config.json +23 -0
speedai-125M-config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GPTModel"
4
+ ],
5
+ "hidden_size": 768,
6
+ "intermediate_size": 3072,
7
+ "num_attention_heads": 12,
8
+ "num_hidden_layers": 12,
9
+ "vocab_size": 50257,
10
+ "max_position_embeddings": 2048,
11
+ "bos_token_id": 0,
12
+ "eos_token_id": 1,
13
+ "pad_token_id": 2,
14
+ "tie_word_embeddings": true,
15
+ "initializer_range": 0.02,
16
+ "layer_norm_epsilon": 1e-05,
17
+ "torch_dtype": "float32",
18
+ "use_cache": true,
19
+ "model_type": "gpt2",
20
+ "attn_pdrop": 0.1,
21
+ "embd_pdrop": 0.1,
22
+ "resid_pdrop": 0.1
23
+ }