persadian commited on
Commit
1295df3
·
verified ·
1 Parent(s): ca3d93e

Update PersadianNanoV4Config.json

Browse files
Files changed (1) hide show
  1. PersadianNanoV4Config.json +25 -38
PersadianNanoV4Config.json CHANGED
@@ -1,49 +1,36 @@
1
- class PersadianNanoV4Config:
2
- def __init__(self):
 
 
3
 
4
- # Core architecture (MATCH CHECKPOINT)
5
- self.hidden_size = 512
6
- self.intermediate_size = 1024
7
- self.num_hidden_layers = 12
8
 
9
- # Attention
10
- self.num_attention_heads = 8
11
- self.num_key_value_heads = 4
12
 
13
- # MoE (MATCH CHECKPOINT)
14
- self.num_experts = 4
15
- self.num_experts_per_tok = 2
16
 
17
- # Progressive experts
18
- self.progressive_experts = True
19
- self.min_experts = 1
20
- self.max_experts = 2
21
 
22
- # Hyper connections
23
- self.use_hyper_connection = True
24
 
25
- # Compression
26
- self.use_compressed_attention = True
27
- self.compress_ratio = 4
28
 
29
- # Context
30
- self.max_position_embeddings = 2048
 
31
 
32
- # Tokenizer
33
- self.vocab_size = 50257
34
- self.bos_token_id = 50256
35
- self.eos_token_id = 50256
36
 
37
- # RoPE
38
- self.rope_theta = 10000.0
39
 
40
- # Regularization
41
- self.dropout = 0.1
42
- self.layer_norm_eps = 1e-5
43
 
44
- # Attention
45
- self.attention_dropout = 0.0
46
- self.use_flash_attention = True
47
-
48
- # Precision (IMPORTANT: keep as attribute, not torch dtype object)
49
- self.torch_dtype = "float16"
 
1
+ {
2
+ "hidden_size": 768,
3
+ "intermediate_size": 2048,
4
+ "num_hidden_layers": 12,
5
 
6
+ "num_attention_heads": 12,
7
+ "num_key_value_heads": 4,
 
 
8
 
9
+ "num_experts": 8,
10
+ "num_experts_per_tok": 2,
 
11
 
12
+ "progressive_experts": true,
13
+ "min_experts": 1,
14
+ "max_experts": 2,
15
 
16
+ "use_hyper_connection": true,
 
 
 
17
 
18
+ "use_compressed_attention": true,
19
+ "compress_ratio": 4,
20
 
21
+ "max_position_embeddings": 2048,
 
 
22
 
23
+ "vocab_size": 50257,
24
+ "bos_token_id": 50256,
25
+ "eos_token_id": 50256,
26
 
27
+ "rope_theta": 10000.0,
 
 
 
28
 
29
+ "dropout": 0.1,
30
+ "layer_norm_eps": 1e-5,
31
 
32
+ "attention_dropout": 0.0,
33
+ "use_flash_attention": true,
 
34
 
35
+ "torch_dtype": "float16"
36
+ }