{ | |
"_class_name": "StableCascadeUNet", | |
"_diffusers_version": "0.27.0.dev0", | |
"block_out_channels": [ | |
320, | |
640, | |
1280, | |
1280 | |
], | |
"block_types_per_layer": [ | |
[ | |
"SDCascadeResBlock", | |
"SDCascadeTimestepBlock" | |
], | |
[ | |
"SDCascadeResBlock", | |
"SDCascadeTimestepBlock" | |
], | |
[ | |
"SDCascadeResBlock", | |
"SDCascadeTimestepBlock", | |
"SDCascadeAttnBlock" | |
], | |
[ | |
"SDCascadeResBlock", | |
"SDCascadeTimestepBlock", | |
"SDCascadeAttnBlock" | |
] | |
], | |
"clip_image_in_channels": null, | |
"clip_seq": 4, | |
"clip_text_in_channels": null, | |
"clip_text_pooled_in_channels": 1280, | |
"conditioning_dim": 1280, | |
"down_blocks_repeat_mappers": [ | |
1, | |
1, | |
1, | |
1 | |
], | |
"down_num_layers_per_block": [ | |
2, | |
6, | |
28, | |
6 | |
], | |
"dropout": [ | |
0, | |
0, | |
0.1, | |
0.1 | |
], | |
"effnet_in_channels": 16, | |
"in_channels": 4, | |
"kernel_size": 3, | |
"num_attention_heads": [ | |
0, | |
0, | |
20, | |
20 | |
], | |
"out_channels": 4, | |
"patch_size": 2, | |
"pixel_mapper_in_channels": 3, | |
"self_attn": true, | |
"switch_level": null, | |
"timestep_conditioning_type": [ | |
"sca" | |
], | |
"timestep_ratio_embedding_dim": 64, | |
"up_blocks_repeat_mappers": [ | |
3, | |
3, | |
2, | |
2 | |
], | |
"up_num_layers_per_block": [ | |
6, | |
28, | |
6, | |
2 | |
] | |
} | |