Compactbot commited on
Commit
e834811
·
verified ·
1 Parent(s): a6ec395

Add architecture config (4,912,992 params)

Browse files
Files changed (1) hide show
  1. config.json +13 -12
config.json CHANGED
@@ -1,16 +1,17 @@
1
  {
2
  "architectures": ["CompactLM"],
3
- "vocab_size": 12288,
4
- "d_model": 256,
5
- "n_layers": 4,
6
- "n_heads": 4,
7
- "head_dim": 64,
8
- "ff_dim": 640,
 
 
9
  "context_length": 512,
 
 
10
  "tie_word_embeddings": true,
11
- "norm_eps": 1e-05,
12
- "rope_base": 10000.0,
13
- "n_params": 6162688,
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
+ }