model_config.json

{
  "vocab_size": 32768,
  "max_position_embeddings": 512,
  "hidden_size": 256,
  "num_attention_heads": 256,
  "num_hidden_layers": 20,
  "num_key_value_heads": 64,
  "head_dim": 1,
  "dtype": "float32",
  "intermediate_size": 2048,
  "mlp_bias": false,
  "rms_norm_eps": 1e-6,
  "bias": false,
  "attention_bias": false
}