ckpts/slat_flow_img2shape_dit_1_3B_1024_bf16.json

{
    "name": "ElasticSLatFlowModel",
    "args": {
        "resolution": 64,
        "in_channels": 32,
        "out_channels": 32,
        "model_channels": 1536,
        "cond_channels": 1024,
        "num_blocks": 30,
        "num_heads": 12,
        "mlp_ratio": 5.3334,
        "pe_mode": "rope",
        "share_mod": true,
        "initialization": "scaled",
        "qk_rms_norm": true,
        "qk_rms_norm_cross": true,
        "image_attn_mode": "proj",
        "proj_in_channels": 2048,
        "dtype": "bfloat16"
    }
}