diffusion_head/config.json

{
  "P_mean": 0.0,
  "P_std": 1.0,
  "_class_name": "BitDanceDiffusionHead",
  "_diffusers_version": "0.36.0",
  "ch_cond": 5120,
  "ch_latent": 5120,
  "ch_target": 32,
  "depth_adanln": 2,
  "depth_latent": 6,
  "diff_batch_mul": 1,
  "grad_checkpointing": false,
  "parallel_num": 16,
  "time_schedule": "logit_normal",
  "time_shift": 1.0,
  "use_swiglu": true
}