config.json

{
  "vocab_size": 12000,
  "block_size": 256,
  "n_layer": 8,
  "n_head": 10,
  "n_embd": 640,
  "user_tok": "<|user|>",
  "assistant_tok": "<|assistant|>",
  "end_tok": "<|end|>"
}