ckpts/shape_enc_next_dc_f16c32_fp16.json

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