ckpts/tex_enc_next_dc_f16c32_fp16.json

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