vision_head_config.json

{
    "ch_target": 32,
    "ch_cond": 5120,
    "ch_latent": 5120,
    "depth_latent": 6,
    "depth_adanln": 2,
    "parallel_num": 64,
    "use_swiglu": true
}