SHAPE_TOKEN_BUDGET = 200_000

SHAPE_CONFIGS = [
    {"label": "wide_shallow", "emb_dim": 128, "n_heads": 4, "n_layers": 2},
    {"label": "balanced",     "emb_dim": 120, "n_heads": 4, "n_layers": 7},
    {"label": "narrow_deep",  "emb_dim": 112, "n_heads": 4, "n_layers": 13},
]

        __
        __