config.json

{
    "model_type": "cosmos",
    "attn_resolutions": [32],
    "channels": 128,
    "channels_mult": [2, 4, 4],
    "dropout": 0.0,
    "in_channels": 3,
    "spatial_compression": 8,
    "num_res_blocks": 2,
    "out_channels": 3,
    "resolution": 1024,
    "patch_size": 4,
    "patch_method": "haar",
    "z_channels": 256,
    "z_factor": 1,
    "quantizer": "FSQ",
    "embedding_dim": 6,
    "levels": [8, 8, 8, 5, 5, 5],
    "num_quantizers": 4,
    "encoder": "Default",
    "decoder": "Default"
}