config.json

{
  "architectures": ["GPT2LMHeadModel"],
  "n_embd": 64,
  "n_head": 4,
  "n_layer": 4,
  "vocab_size": 61,
  "block_size": 32,
  "dropout": 0.1,
  "model_type": "gpt2"
}