config.json

{
  "_attn_implementation_autoset": true,
  "block_size": 1024,
  "dropout": 0.2,
  "model_type": "gpt_custom",
  "n_embd": 768,
  "n_head": 12,
  "n_layer": 12,
  "transformers_version": "4.46.2",
  "vocab_size": 20000
}