config.json

{
  "input_dim": 1280,
  "hidden_dim": 1280,
  "mlp_ratio": 2.0,
  "num_heads": 20,
  "num_layers": 20,
  "conditioning_dropout": 0.1
}