Diffusers documentation
StableCascadeUNet
StableCascadeUNet
نموذج UNet من خط أنابيب Stable Cascade.
StableCascadeUNet
فئة diffusers.models.StableCascadeUNet
< source >( 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
< source >( 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 عادي.