config.json

{
  "model_type": "custom-gpt",
  "vocab_size": 50257,
  "n_layer": 6,
  "n_head": 6,
  "n_embd": 384,
  "block_size": 128,
  "dropout": 0.1,
  "bias": true,
  "architectural_notes": "Notebook custom GPT implementation. Not compatible with transformers AutoModel!"
}