ckpts/slat_flow_txt_dit_B_64l8p2_fp16.json

{
    "name": "SLatFlowModel",
    "args": {
        "resolution": 64,
        "in_channels": 8,
        "out_channels": 8,
        "model_channels": 768,
        "cond_channels": 768,
        "num_blocks": 12,
        "num_heads": 12,
        "mlp_ratio": 4,
        "patch_size": 2,
        "num_io_res_blocks": 2,
        "io_block_channels": [128],
        "pe_mode": "ape",
        "qk_rms_norm": true,
        "use_fp16": true
    }
}