{
  "_class_name": "StableAudioDiT",
  "io_channels": 64,
  "embed_dim": 1024,
  "depth": 16,
  "cond_token_dim": 768,
  "global_cond_dim": 768,
  "num_attention_heads": 8,
  "qk_norm": "ln",
  "project_cond_tokens": true,
  "project_global_cond": true
}