| { | |
| "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 | |
| } | |
| } |