{ | |
"_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] | |
} |