config.json

{
  "model_type": "gpt2",
  "vocab_size": 8000,
  "n_positions": 512,
  "n_ctx": 512,
  "n_embd": 256,
  "n_layer": 4,
  "n_head": 4
}