stage2_slat_flow.json

{
  "name": "TriSLatSeparateFlowModel",
  "args": {
    "resolution": 32,
    "in_channels": 32,
    "out_channels": 32,
    "model_channels": 1536,
    "cond_channels": 1024,
    "num_blocks": 30,
    "num_heads": 12,
    "mlp_ratio": 5.3334,
    "share_mod": true,
    "qk_rms_norm": true,
    "qk_rms_norm_cross": true,
    "global_attn_at": [
      2,
      5,
      8,
      11,
      14,
      17,
      20,
      23,
      26,
      29
    ],
    "num_streams": 2
  }
}