| { | |
| "name": "FlexiDualGridVaeEncoder", | |
| "args": { | |
| "model_channels": [64, 128, 256, 512, 1024], | |
| "latent_channels": 32, | |
| "num_blocks": [0, 4, 8, 16, 4], | |
| "block_type": [ | |
| "SparseConvNeXtBlock3d", | |
| "SparseConvNeXtBlock3d", | |
| "SparseConvNeXtBlock3d", | |
| "SparseConvNeXtBlock3d", | |
| "SparseConvNeXtBlock3d" | |
| ], | |
| "down_block_type": [ | |
| "SparseResBlockS2C3d", | |
| "SparseResBlockS2C3d", | |
| "SparseResBlockS2C3d", | |
| "SparseResBlockS2C3d" | |
| ], | |
| "block_args": [{}, {}, {}, {}, {}], | |
| "use_fp16": true | |
| } | |
| } |