Attention Processor

Attention Processor

An attention processor is a class for applying different types of attention mechanisms.

AttnProcessor

class diffusers.models.attention_processor.AttnProcessor

<source>

( )

Default processor for performing attention-related computations.

AttnProcessor2_0

class diffusers.models.attention_processor.AttnProcessor2_0

<source>

( )

Processor for implementing scaled dot-product attention (enabled by default if you’re using PyTorch 2.0).

LoRAAttnProcessor

class diffusers.models.attention_processor.LoRAAttnProcessor

<source>

( hidden_sizecross_attention_dim = Nonerank = 4network_alpha = None**kwargs )

Parameters

  • hidden_size (int, optional) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional) — The number of channels in the encoder_hidden_states.

  • rank (int, defaults to 4) — The dimension of the LoRA update matrices.

  • network_alpha (int, optional) — Equivalent to alpha but it’s usage is specific to Kohya (A1111) style LoRAs.

Processor for implementing the LoRA attention mechanism.

LoRAAttnProcessor2_0

class diffusers.models.attention_processor.LoRAAttnProcessor2_0

<source>

( hidden_sizecross_attention_dim = Nonerank = 4network_alpha = None**kwargs )

Parameters

  • hidden_size (int) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional) — The number of channels in the encoder_hidden_states.

  • rank (int, defaults to 4) — The dimension of the LoRA update matrices.

  • network_alpha (int, optional) — Equivalent to alpha but it’s usage is specific to Kohya (A1111) style LoRAs.

Processor for implementing the LoRA attention mechanism using PyTorch 2.0’s memory-efficient scaled dot-product attention.

CustomDiffusionAttnProcessor

class diffusers.models.attention_processor.CustomDiffusionAttnProcessor

<source>

( train_kv = Truetrain_q_out = Truehidden_size = Nonecross_attention_dim = Noneout_bias = Truedropout = 0.0 )

Parameters

  • train_kv (bool, defaults to True) — Whether to newly train the key and value matrices corresponding to the text features.

  • train_q_out (bool, defaults to True) — Whether to newly train query matrices corresponding to the latent image features.

  • hidden_size (int, optional, defaults to None) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional, defaults to None) — The number of channels in the encoder_hidden_states.

  • out_bias (bool, defaults to True) — Whether to include the bias parameter in train_q_out.

  • dropout (float, optional, defaults to 0.0) — The dropout probability to use.

Processor for implementing attention for the Custom Diffusion method.

AttnAddedKVProcessor

class diffusers.models.attention_processor.AttnAddedKVProcessor

<source>

( )

Processor for performing attention-related computations with extra learnable key and value matrices for the text encoder.

AttnAddedKVProcessor2_0

class diffusers.models.attention_processor.AttnAddedKVProcessor2_0

<source>

( )

Processor for performing scaled dot-product attention (enabled by default if you’re using PyTorch 2.0), with extra learnable key and value matrices for the text encoder.

LoRAAttnAddedKVProcessor

class diffusers.models.attention_processor.LoRAAttnAddedKVProcessor

<source>

( hidden_sizecross_attention_dim = Nonerank = 4network_alpha = None )

Parameters

  • hidden_size (int, optional) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional, defaults to None) — The number of channels in the encoder_hidden_states.

  • rank (int, defaults to 4) — The dimension of the LoRA update matrices.

Processor for implementing the LoRA attention mechanism with extra learnable key and value matrices for the text encoder.

XFormersAttnProcessor

class diffusers.models.attention_processor.XFormersAttnProcessor

<source>

( attention_op: typing.Optional[typing.Callable] = None )

Parameters

  • attention_op (Callable, optional, defaults to None) — The base operator to use as the attention operator. It is recommended to set to None, and allow xFormers to choose the best operator.

Processor for implementing memory efficient attention using xFormers.

LoRAXFormersAttnProcessor

class diffusers.models.attention_processor.LoRAXFormersAttnProcessor

<source>

( hidden_sizecross_attention_dimrank = 4attention_op: typing.Optional[typing.Callable] = Nonenetwork_alpha = None**kwargs )

Parameters

  • hidden_size (int, optional) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional) — The number of channels in the encoder_hidden_states.

  • rank (int, defaults to 4) — The dimension of the LoRA update matrices.

  • attention_op (Callable, optional, defaults to None) — The base operator to use as the attention operator. It is recommended to set to None, and allow xFormers to choose the best operator.

  • network_alpha (int, optional) — Equivalent to alpha but it’s usage is specific to Kohya (A1111) style LoRAs.

Processor for implementing the LoRA attention mechanism with memory efficient attention using xFormers.

CustomDiffusionXFormersAttnProcessor

class diffusers.models.attention_processor.CustomDiffusionXFormersAttnProcessor

<source>

( train_kv = Truetrain_q_out = Falsehidden_size = Nonecross_attention_dim = Noneout_bias = Truedropout = 0.0attention_op: typing.Optional[typing.Callable] = None )

Parameters

  • train_kv (bool, defaults to True) — Whether to newly train the key and value matrices corresponding to the text features.

  • train_q_out (bool, defaults to True) — Whether to newly train query matrices corresponding to the latent image features.

  • hidden_size (int, optional, defaults to None) — The hidden size of the attention layer.

  • cross_attention_dim (int, optional, defaults to None) — The number of channels in the encoder_hidden_states.

  • out_bias (bool, defaults to True) — Whether to include the bias parameter in train_q_out.

  • dropout (float, optional, defaults to 0.0) — The dropout probability to use.

  • attention_op (Callable, optional, defaults to None) — The base operator to use as the attention operator. It is recommended to set to None, and allow xFormers to choose the best operator.

Processor for implementing memory efficient attention using xFormers for the Custom Diffusion method.

SlicedAttnProcessor

class diffusers.models.attention_processor.SlicedAttnProcessor

<source>

( slice_size )

Parameters

  • slice_size (int, optional) — The number of steps to compute attention. Uses as many slices as attention_head_dim // slice_size, and attention_head_dim must be a multiple of the slice_size.

Processor for implementing sliced attention.

SlicedAttnAddedKVProcessor

class diffusers.models.attention_processor.SlicedAttnAddedKVProcessor

<source>

( slice_size )

Parameters

  • slice_size (int, optional) — The number of steps to compute attention. Uses as many slices as attention_head_dim // slice_size, and attention_head_dim must be a multiple of the slice_size.

Processor for implementing sliced attention with extra learnable key and value matrices for the text encoder.

Last updated