config.json

{
  "vocab_size": 50257,
  "block_size": 128,
  "n_layer": 6,
  "n_head": 6,
  "n_embd": 384,
  "dropout": 0.1,
  "bias": true,
  "_name_or_path": "custom-gpt",
  "model_type": "custom-gpt"
 
}