Jtapsa commited on
Commit
c0e18e5
·
verified ·
1 Parent(s): cc41612

Add config.json

Browse files
Files changed (1) hide show
  1. config.json +33 -0
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["MoEPForCausalLM"],
3
+ "auto_map": {
4
+ "AutoConfig": "moep.MoEPLMConfig",
5
+ "AutoModelForCausalLM": "moep.MoEPForCausalLM"
6
+ }
7
+ ,
8
+ "attn_pdrop": 0.1,
9
+ "bos_token_id": 1,
10
+ "dim": 384,
11
+ "embd_pdrop": 0.1,
12
+ "eos_token_id": 2,
13
+ "ffn_multp": 4,
14
+ "layer_norm_epsilon": 1e-05,
15
+ "mask_token_id": 4,
16
+ "max_position_embeddings": 1024,
17
+ "model_type": "moep_lm",
18
+ "moe_k": 2,
19
+ "n_experts": 4,
20
+ "n_heads": 6,
21
+ "n_layers": 12,
22
+ "n_parallel_blocks": 4,
23
+ "pad_token_id": 3,
24
+ "parallel_dim": 192,
25
+ "parallel_k": 2,
26
+ "parallel_n_heads": 3,
27
+ "resid_pdrop": 0.1,
28
+ "sliding_window": null,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.51.3",
31
+ "vocab_size": 16384,
32
+ "weight_tyingtrue": true
33
+ }