Diffusers documentation
Слои нормализации
Слои нормализации
Специализированные слои нормализации для поддержки различных моделей в 🤗 Diffusers.
AdaLayerNorm
класс diffusers.models.normalization.AdaLayerNorm
< source >( embedding_dim: intnum_embeddings: int | None = Noneoutput_dim: int | None = Nonenorm_elementwise_affine: bool = Falsenorm_eps: float = 1e-05chunk_dim: int = 0 )
Слой нормализации изменён для включения эмбеддингов временных шагов.
AdaLayerNormZero
класс diffusers.models.normalization.AdaLayerNormZero
< source >( embedding_dim: intnum_embeddings: int | None = Nonenorm_type = 'layer_norm'bias = True )
Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).
AdaLayerNormSingle
класс diffusers.models.normalization.AdaLayerNormSingle
< source >( embedding_dim: intuse_additional_conditions: bool = False )
Нормализующий слой адаптивная нормализация по слоям single (adaLN-single).
Как предложено в PixArt-Alpha (см.: https://huggingface.co/papers/2310.00426; раздел 2.3).
AdaGroupNorm
класс diffusers.models.normalization.AdaGroupNorm
< source >( embedding_dim: intout_dim: intnum_groups: intact_fn: str | None = Noneeps: float = 1e-05 )
Параметры
- embedding_dim (
int) — размерность каждого вектора эмбеддинга. - num_embeddings (
int) — Размер словаря эмбеддингов. - num_groups (
int) — Количество групп, на которые разделяются каналы. - act_fn (
str, необязательный, по умолчаниюNone) — Функция активации, которую следует использовать. - eps (
float, необязательный, по умолчанию1e-5) — значение эпсилона, используемое для численной устойчивости.
Слой GroupNorm, модифицированный для включения эмбеддингов временных шагов.
AdaLayerNormContinuous
класс diffusers.models.normalization.AdaLayerNormContinuous
< source >( embedding_dim: intconditioning_embedding_dim: intelementwise_affine = Trueeps = 1e-05bias = Truenorm_type = 'layer_norm' )
Параметры
- embedding_dim (
int) — Размерность эмбеддинга, используемая при проекции. - conditioning_embedding_dim (
int) — Размерность входного условия. - elementwise_affine (
bool, по умолчаниюTrue) — Логический флаг, указывающий, следует ли применять аффинное преобразование. - eps (
float, по умолчанию 1e-5) — Коэффициент эпсилон. - bias (
bias, по умолчаниюTrue) — булевый флаг, указывающий, следует ли использовать смещение. - norm_type (
str, по умолчанию"layer_norm") — Слой нормализации для использования. Поддерживаемые значения: “layer_norm”, “rms_norm”.
Слой адаптивной нормализации со слоем нормализации (layer_norm или rms_norm).
RMSNorm
класс diffusers.models.normalization.RMSNorm
< source >( dimeps: floatelementwise_affine: bool = Truebias: bool = False )
Параметры
- dim (
int) — Количество измерений дляweights. Действует только приelementwise_affine= True. - eps (
float) — Малое значение, используемое при вычислении обратной величины квадратного корня. - elementwise_affine (
bool, по умолчаниюTrue) — Логический флаг, указывающий, следует ли применять аффинное преобразование. - bias (
bool, по умолчанию False) — Если также обучать параметрbias.
RMS Norm, предложенный в https://huggingface.co/papers/1910.07467 Zhang et al.
GlobalResponseNorm
класс diffusers.models.normalization.GlobalResponseNorm
< source >( dim )
Нормализация глобального ответа, предложенная в ConvNeXt-v2 (https://huggingface.co/papers/2301.00808).
LuminaLayerNormContinuous
класс diffusers.models.normalization.LuminaLayerNormContinuous
< source >( embedding_dim: intconditioning_embedding_dim: intelementwise_affine = Trueeps = 1e-05bias = Truenorm_type = 'layer_norm'out_dim: int | None = None )
SD35AdaLayerNormZeroX
класс diffusers.models.normalization.SD35AdaLayerNormZeroX
< source >( embedding_dim: intnorm_type: str = 'layer_norm'bias: bool = True )
Нормализующий слой адаптивной нормализации нулей (AdaLN-Zero).
AdaLayerNormZeroSingle
класс diffusers.models.normalization.AdaLayerNormZeroSingle
< source >( embedding_dim: intnorm_type = 'layer_norm'bias = True )
Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).
LuminaRMSNormZero
класс diffusers.models.normalization.LuminaRMSNormZero
< source >( embedding_dim: intnorm_eps: floatnorm_elementwise_affine: bool )
Слой нормализации адаптивная RMS-нормализация ноль.
LpNorm
класс diffusers.models.normalization.LpNorm
< source >( p: int = 2dim: int = -1eps: float = 1e-12 )
CogView3PlusAdaLayerNormZeroTextImage
класс diffusers.models.normalization.CogView3PlusAdaLayerNormZeroTextImage
< source >( embedding_dim: intdim: int )
Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).
CogVideoXLayerNormZero
класс diffusers.models.normalization.CogVideoXLayerNormZero
< source >( conditioning_dim: intembedding_dim: intelementwise_affine: bool = Trueeps: float = 1e-05bias: bool = True )
MochiRMSNormZero
класс diffusers.models.transformers.transformer_mochi.MochiRMSNormZero
< source >( embedding_dim: inthidden_dim: inteps: float = 1e-05elementwise_affine: bool = False )
Адаптивная RMS Norm, используемая в Mochi.
MochiRMSNorm
класс diffusers.models.normalization.MochiRMSNorm
< source >( dimeps: floatelementwise_affine: bool = True )