config.json

{
  "depth": 4,
  "dim": 256,
  "embed_dim": 128,
  "heads": 8,
  "hidden": 1024
}