{
  "fused_dim": 2048,
  "hidden_dim": 2048,
  "num_tokens": 64,
  "embed_dim": 1536
}