| { | |
| "name": "SparseStructureFlowModel", | |
| "args":{ | |
| "resolution": 16, | |
| "in_channels": 32, | |
| "out_channels": 32, | |
| "model_channels": 1280, | |
| "cond_channels": 1280, | |
| "num_blocks": 36, | |
| "num_heads": 32, | |
| "mlp_ratio": 4, | |
| "patch_size": 1, | |
| "pe_mode": "ape", | |
| "qk_rms_norm": true, | |
| "use_fp16": true | |
| } | |
| } |