ckpts/tex_dec_next_dc_f16c32_fp16.json

{
    "name": "SparseUnetVaeDecoder",
    "args": {
        "out_channels": 6,
        "model_channels": [1024, 512, 256, 128, 64],
        "latent_channels": 32,
        "num_blocks": [4, 16, 8, 4, 0],
        "block_type": [
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d",
            "SparseConvNeXtBlock3d"
        ],
        "up_block_type": [
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d",
            "SparseResBlockC2S3d"
        ],
        "block_args": [{}, {}, {}, {}, {}],
        "pred_subdiv": false,
        "use_fp16": true
    }
}