Diffusers documentation
StableCascadeUNet
StableCascadeUNet
A UNet model from the Stable Cascade pipeline.
StableCascadeUNet
class diffusers.models.StableCascadeUNet
< source >( in_channels: int = 16 out_channels: int = 16 timestep_ratio_embedding_dim: int = 64 patch_size: int = 1 conditioning_dim: int = 2048 block_out_channels: tuple = (2048, 2048) num_attention_heads: tuple = (32, 32) down_num_layers_per_block: tuple = (8, 24) up_num_layers_per_block: tuple = (24, 8) down_blocks_repeat_mappers: tuple[int] | None = (1, 1) up_blocks_repeat_mappers: tuple[int] | None = (1, 1) block_types_per_layer: tuple = (('SDCascadeResBlock', 'SDCascadeTimestepBlock', 'SDCascadeAttnBlock'), ('SDCascadeResBlock', 'SDCascadeTimestepBlock', 'SDCascadeAttnBlock')) clip_text_in_channels: int | None = None clip_text_pooled_in_channels = 1280 clip_image_in_channels: int | None = None clip_seq = 4 effnet_in_channels: int | None = None pixel_mapper_in_channels: int | None = None kernel_size = 3 dropout: float | tuple[float] = (0.1, 0.1) self_attn: bool | tuple[bool] = True timestep_conditioning_type: tuple = ('sca', 'crp') switch_level: tuple[bool] | None = None )