{
  "feat_dim": 8448,
  "num_clusters": 64,
  "cluster_dim": 256,
  "token_dim": 256,
  "mlp_dim": 512
}