config.json

{
  "channels": [
    128,
    256,
    512,
    1024
  ],
  "midcoder_dropout_p": 0.2,
  "num_residual_layers": 2,
  "t_embed_dim": 128
}