config.json

{
  "small": {
    "d_model": 128,
    "hidden": 1024
  }
}