config.json

{"d_model": 128, "n_heads": 4, "n_layers": 2, "d_ff": 256, "dropout": 0.3, "src_vocab_size": 5000, "trg_vocab_size": 5000}