{
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.17.0.dev0",
   "model_channels": 384,
   "num_channels": 4,
   "init_channels": 192,
   "time_embed_dim": 1536,
   "context_dim": 4096,
   "model_dim": 4096,
   "groups": 32,
   "head_dim": 64,
   "expansion_ratio": 4,
   "compression_ratio": 2,
   "dim_mult": [1, 2, 4, 8],
   "num_blocks": [3, 3, 3, 3],
   "add_cross_attention": [false, true, true, true],
   "add_self_attention": [false, true, true, true]
}