Diffusers documentation

StableCascadeUNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

StableCascadeUNet

نموذج UNet من خط أنابيب Stable Cascade.

StableCascadeUNet

فئة diffusers.models.StableCascadeUNet

< >

( in_channels: int = 16out_channels: int = 16timestep_ratio_embedding_dim: int = 64patch_size: int = 1conditioning_dim: int = 2048block_out_channels: tuple[int, ...] = (2048, 2048)num_attention_heads: tuple[int, ...] = (32, 32)down_num_layers_per_block: tuple[int, ...] = (8, 24)up_num_layers_per_block: tuple[int, ...] = (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[tuple[str]] = (('SDCascadeResBlock', 'SDCascadeTimestepBlock', 'SDCascadeAttnBlock'), ('SDCascadeResBlock', 'SDCascadeTimestepBlock', 'SDCascadeAttnBlock'))clip_text_in_channels: int | None = Noneclip_text_pooled_in_channels = 1280clip_image_in_channels: int | None = Noneclip_seq = 4effnet_in_channels: int | None = Nonepixel_mapper_in_channels: int | None = Nonekernel_size = 3dropout: float | tuple[float] = (0.1, 0.1)self_attn: bool | tuple[bool] = Truetimestep_conditioning_type: tuple[str, ...] = ('sca', 'crp')switch_level: tuple[bool] | None = None )

forward

< >

( sampletimestep_ratioclip_text_pooledclip_text = Noneclip_img = Noneeffnet = Nonepixels = Nonesca = Nonecrp = Nonereturn_dict = True )

المعاملات

  • sample (torch.Tensor) — عينة الإدخال الضجيجية.
  • timestep_ratio (torch.Tensor) — نسبة الخطوة الزمنية المستخدمة لحساب تضمين الخطوة الزمنية.
  • clip_text_pooled (torch.Tensor) — تضمينات نص CLIP المجمّعة.
  • clip_text (torch.Tensor، اختياري) — تضمينات نص CLIP على مستوى التسلسل.
  • clip_img (torch.Tensor، اختياري) — تضمينات صور CLIP.
  • effnet (torch.Tensor، اختياري) — خريطة ميزات EfficientNet تُستخدم كتشكيل إضافي.
  • pixels (torch.Tensor، اختياري) — مصفوفة شروط على مستوى البكسل. إذا كانت None، تُستخدم مصفوفة من الأصفار.
  • sca (torch.Tensor, اختياري) — قيمة sca الاختيارية المستخدمة لبناء تضمين خطوة الزمن.
  • crp (torch.Tensor، اختياري) — قيمة crp الاختيارية المستخدمة كشرط لبناء تضمين خطوة الزمن.
  • return_dict (bool، اختياري، القيمة الافتراضية True) — ما إذا كان يجب إرجاع StableCascadeUNetOutput بدلاً من tuple عادي.
Update on GitHub