config.json

{"embed_dim": 256, "num_heads": 4}