Hugging Face's logo

Diffusers 文档

规范化层

Hugging Face's logo
加入 Hugging Face 社区

并获取增强版文档体验

以开始使用

归一化层

用于支持 🤗 Diffusers 中各种模型的自定义归一化层。

AdaLayerNorm

class diffusers.models.normalization.AdaLayerNorm

< >

( embedding_dim: int num_embeddings: Optional = None output_dim: Optional = None norm_elementwise_affine: bool = False norm_eps: float = 1e-05 chunk_dim: int = 0 )

参数

  • embedding_dim (int) — 每个嵌入向量的尺寸。
  • num_embeddings (int, 可选) — 嵌入字典的大小。
  • output_dim (int, 可选) —
  • norm_elementwise_affine (bool, 默认值为 `False) —
  • norm_eps (bool, 默认值为 False) —
  • chunk_dim (int, defaults to 0) —

修改后的规范层，用于合并时间步嵌入。

AdaLayerNormZero

class diffusers.models.normalization.AdaLayerNormZero

< >

( embedding_dim: int num_embeddings: Optional = None norm_type = 'layer_norm' bias = True )

参数

  • embedding_dim (int) — 每个嵌入向量的尺寸。
  • num_embeddings (int) — 嵌入字典的大小。

自适应层规范零 (adaLN-Zero) 的规范层。

AdaLayerNormSingle

class diffusers.models.normalization.AdaLayerNormSingle

< >

( embedding_dim: int use_additional_conditions: bool = False )

参数

  • embedding_dim (int) — 每个嵌入向量的尺寸。
  • use_additional_conditions (bool) — 是否使用额外的条件进行规范化。

自适应层规范单 (adaLN-single) 的规范层。

如 PixArt-Alpha 中所述（参见：https://arxiv.org/abs/2310.00426; 第 2.3 节）。

AdaGroupNorm

< >

( embedding_dim: int out_dim: int num_groups: int act_fn: Optional = None eps: float = 1e-05 )

参数

  • embedding_dim (int) — 每个嵌入向量的尺寸。
  • num_embeddings (int) — 嵌入字典的大小。
  • num_groups (int) — 将通道分成组的数量。
  • act_fn (str, 可选, 默认值为 None) — 要使用的激活函数。
  • eps (float, 可选, 默认值为 1e-5) — 用于数值稳定性的 epsilon 值。

GroupNorm 层经过修改，以整合时间步长嵌入。

< > 在 GitHub 上更新

←自定义激活函数 实用工具→