config.json

{
  "diffusion_step_embed_dim_in": 128,
  "diffusion_step_embed_dim_mid": 512,
  "diffusion_step_embed_dim_out": 512,
  "in_channels": 8,
  "label_embed_classes": 71,
  "label_embed_dim": 128,
  "num_res_layers": 36,
  "out_channels": 8,
  "res_channels": 256,
  "s4_bidirectional": 1,
  "s4_d_state": 64,
  "s4_dropout": 0.0,
  "s4_layernorm": 1,
  "s4_lmax": 1000,
  "skip_channels": 256
}