sparse_structure_flow_model/model.json

{
  "name": "SparseStructureFlowModel",
  "args": {
    "patch_size": 1,
    "resolution": 16,
    "use_checkpoint": false,
    "pe_mode": "ape",
    "num_blocks": 24,
    "cond_channels": 1024,
    "share_mod": false,
    "mlp_ratio": 4,
    "qk_rms_norm": true,
    "qk_rms_norm_cross": false,
    "model_channels": 1024,
    "num_heads": 16,
    "in_channels": 8,
    "use_fp16": true,
    "out_channels": 8
  }
}