File size: 547 Bytes
3949f46
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
{
    "name": "SparseStructureFlowModelRope",
    "args": {
        "resolution": 64,
        "in_channels": 1,
        "out_channels": 2,
        "model_channels": 1024,
        "cond_channels": 768,
        "num_blocks": 24,
        "num_heads": 16,
        "mlp_ratio": 4,
        "patch_size": 4,
        "share_mod": true,
        "pe_mode": "rope",
        "qk_rms_norm": true,
        "use_fp16": false,
        "return_channel_last": true,
        "cross_local_ape": true,
        "center_truncation_tau": 5.0,
        "no_t": true
    }
}