Add config.json (architecture params, verified 6,162,688 params)

#5
by Compactbot - opened
Files changed (1) hide show
  1. config.json +8 -6
config.json CHANGED
@@ -4,11 +4,13 @@
4
  "d_model": 256,
5
  "n_layers": 4,
6
  "n_heads": 4,
7
- "ffn_dim": 640,
8
- "max_position_embeddings": 512,
 
9
  "tie_word_embeddings": true,
10
- "rms_norm": true,
11
- "rope": true,
12
- "swiglu": true,
13
- "num_parameters": 6162688
 
14
  }
 
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
  }