config.json

{
  "model_type": "transformer_lm",
  "architectures": [
    "TransformerLM"
  ],
  "vocab_size": 30000,
  "context_length": 512,
  "num_layers": 8,
  "d_model": 768,
  "num_heads": 12,
  "d_ff": 2048,
  "theta": 10000.0
}