Cytrex commited on
Commit
d93d8a9
·
verified ·
1 Parent(s): 6e231be

Add 26B-A4B MoE config (hidden=2816, head_dim=256, 16 heads)

Browse files
scripts/hf_job_selfdistill_train.py CHANGED
@@ -88,6 +88,14 @@ GEMMA4_CONFIGS = {
88
  "num_key_value_heads": 2,
89
  "vocab_size": 262144,
90
  },
 
 
 
 
 
 
 
 
91
  "27b": {
92
  "hidden_size": 3584,
93
  "intermediate_size": 14336,
 
88
  "num_key_value_heads": 2,
89
  "vocab_size": 262144,
90
  },
91
+ "26b": {
92
+ "hidden_size": 2816,
93
+ "intermediate_size": 2112,
94
+ "num_attention_heads": 16,
95
+ "num_key_value_heads": 8,
96
+ "vocab_size": 262144,
97
+ "head_dim": 256,
98
+ },
99
  "27b": {
100
  "hidden_size": 3584,
101
  "intermediate_size": 14336,