weights/encoder.json

{
    "name": "UniLatEncoder",
    "args": {
        "resolution": 64,
        "in_channels": 1024,
        "model_channels": 768,
        "latent_channels": 8,
        "num_blocks": 12,
        "num_heads": 12,
        "mlp_ratio": 4,
        "attn_mode": "swin",
        "window_size": 8,
        "use_fp16": true,
        "feats_latent_channels": 32,
        "num_res_blocks": 4,
        "pe_voxel_dim": 0
    }
}