Add architecture config (4,912,992 params)
Browse files- config.json +13 -12
config.json
CHANGED
|
@@ -1,16 +1,17 @@
|
|
| 1 |
{
|
| 2 |
"architectures": ["CompactLM"],
|
| 3 |
-
"
|
| 4 |
-
"
|
| 5 |
-
"
|
| 6 |
-
"
|
| 7 |
-
"
|
| 8 |
-
"
|
|
|
|
|
|
|
| 9 |
"context_length": 512,
|
|
|
|
|
|
|
| 10 |
"tie_word_embeddings": true,
|
| 11 |
-
"
|
| 12 |
-
"
|
| 13 |
-
|
| 14 |
-
"dtype": "float32",
|
| 15 |
-
"note": "Custom LLaMA-style architecture (RoPE + causal SDPA + SwiGLU + RMSNorm pre-norm). Not transformers-native; load via the provided train_compactlm5m.py CompactLM class or a matching custom loader."
|
| 16 |
-
}
|
|
|
|
| 1 |
{
|
| 2 |
"architectures": ["CompactLM"],
|
| 3 |
+
"model_type": "compactlm",
|
| 4 |
+
"vocab_size": 8192,
|
| 5 |
+
"hidden_size": 224,
|
| 6 |
+
"intermediate_size": 576,
|
| 7 |
+
"num_hidden_layers": 6,
|
| 8 |
+
"num_attention_heads": 7,
|
| 9 |
+
"num_key_value_heads": 2,
|
| 10 |
+
"head_dim": 32,
|
| 11 |
"context_length": 512,
|
| 12 |
+
"rms_norm_eps": 1e-6,
|
| 13 |
+
"rope_theta": 10000.0,
|
| 14 |
"tie_word_embeddings": true,
|
| 15 |
+
"torch_dtype": "float32",
|
| 16 |
+
"num_parameters": 4912992
|
| 17 |
+
}
|
|
|
|
|
|
|
|
|