| { | |
| "_class_name": "StableCascadeUNet", | |
| "_diffusers_version": "0.27.0.dev0", | |
| "block_out_channels": [ | |
| 320, | |
| 576, | |
| 1152, | |
| 1152 | |
| ], | |
| "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, | |
| 4, | |
| 14, | |
| 4 | |
| ], | |
| "dropout": [ | |
| 0, | |
| 0, | |
| 0.1, | |
| 0.1 | |
| ], | |
| "effnet_in_channels": 16, | |
| "in_channels": 4, | |
| "kernel_size": 3, | |
| "num_attention_heads": [ | |
| 0, | |
| 9, | |
| 18, | |
| 18 | |
| ], | |
| "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": [ | |
| 2, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "up_num_layers_per_block": [ | |
| 4, | |
| 14, | |
| 4, | |
| 2 | |
| ] | |
| } | |