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,
|