config.json

{
    "vocab_size": 8194,
    "num_layers": 4,
    "units": 512,
    "d_model": 128,
    "num_heads": 8,
    "dropout": 0.1,
    "max_length": 40,
    "start_token": [8192],
    "end_token": [8193]
}