File size: 484 Bytes
7d88d1f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
{
"_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]
} |