Diffusers documentation

Слои нормализации

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.42.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Слои нормализации

Специализированные слои нормализации для поддержки различных моделей в 🤗 Diffusers.

AdaLayerNorm

класс diffusers.models.normalization.AdaLayerNorm

< >

( embedding_dim: intnum_embeddings: int | None = Noneoutput_dim: int | None = Nonenorm_elementwise_affine: bool = Falsenorm_eps: float = 1e-05chunk_dim: int = 0 )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • num_embeddings (int, необязательный) — Размер словаря эмбеддингов.
  • output_dim (int, необязательный) —
  • norm_elementwise_affine (bool, по умолчанию `False) —
  • norm_eps (bool, по умолчанию False) —
  • chunk_dim (int, по умолчанию 0) —

Слой нормализации изменён для включения эмбеддингов временных шагов.

AdaLayerNormZero

класс diffusers.models.normalization.AdaLayerNormZero

< >

( embedding_dim: intnum_embeddings: int | None = Nonenorm_type = 'layer_norm'bias = True )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • num_embeddings (int) — Размер словаря эмбеддингов.

Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).

AdaLayerNormSingle

класс diffusers.models.normalization.AdaLayerNormSingle

< >

( embedding_dim: intuse_additional_conditions: bool = False )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • use_additional_conditions (bool) — Использовать дополнительные условия для нормализации или нет.

Нормализующий слой адаптивная нормализация по слоям single (adaLN-single).

Как предложено в PixArt-Alpha (см.: https://huggingface.co/papers/2310.00426; раздел 2.3).

AdaGroupNorm

класс diffusers.models.normalization.AdaGroupNorm

< >

( 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

< >

( 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

< >

( 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

< >

( dim )

Параметры

  • dim (int) — Количество измерений, используемых для gamma и beta.

Нормализация глобального ответа, предложенная в ConvNeXt-v2 (https://huggingface.co/papers/2301.00808).

LuminaLayerNormContinuous

класс diffusers.models.normalization.LuminaLayerNormContinuous

< >

( embedding_dim: intconditioning_embedding_dim: intelementwise_affine = Trueeps = 1e-05bias = Truenorm_type = 'layer_norm'out_dim: int | None = None )

SD35AdaLayerNormZeroX

класс diffusers.models.normalization.SD35AdaLayerNormZeroX

< >

( embedding_dim: intnorm_type: str = 'layer_norm'bias: bool = True )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • num_embeddings (int) — Размер словаря эмбеддингов.

Нормализующий слой адаптивной нормализации нулей (AdaLN-Zero).

AdaLayerNormZeroSingle

класс diffusers.models.normalization.AdaLayerNormZeroSingle

< >

( embedding_dim: intnorm_type = 'layer_norm'bias = True )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • num_embeddings (int) — Размер словаря эмбеддингов.

Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).

LuminaRMSNormZero

класс diffusers.models.normalization.LuminaRMSNormZero

< >

( embedding_dim: intnorm_eps: floatnorm_elementwise_affine: bool )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.

Слой нормализации адаптивная RMS-нормализация ноль.

LpNorm

класс diffusers.models.normalization.LpNorm

< >

( p: int = 2dim: int = -1eps: float = 1e-12 )

CogView3PlusAdaLayerNormZeroTextImage

класс diffusers.models.normalization.CogView3PlusAdaLayerNormZeroTextImage

< >

( embedding_dim: intdim: int )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.
  • num_embeddings (int) — Размер словаря эмбеддингов.

Нормализующий слой адаптивная нормализация нулей (adaLN-Zero).

CogVideoXLayerNormZero

класс diffusers.models.normalization.CogVideoXLayerNormZero

< >

( conditioning_dim: intembedding_dim: intelementwise_affine: bool = Trueeps: float = 1e-05bias: bool = True )

MochiRMSNormZero

класс diffusers.models.transformers.transformer_mochi.MochiRMSNormZero

< >

( embedding_dim: inthidden_dim: inteps: float = 1e-05elementwise_affine: bool = False )

Параметры

  • embedding_dim (int) — размерность каждого вектора эмбеддинга.

Адаптивная RMS Norm, используемая в Mochi.

MochiRMSNorm

класс diffusers.models.normalization.MochiRMSNorm

< >

( dimeps: floatelementwise_affine: bool = True )

Update on GitHub