ShadeEngine's picture
End of training
4ac8f3e verified

Attention Processor

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

AttnProcessor

[[autodoc]] models.attention_processor.AttnProcessor

AttnProcessor2_0

[[autodoc]] models.attention_processor.AttnProcessor2_0

AttnAddedKVProcessor

[[autodoc]] models.attention_processor.AttnAddedKVProcessor

AttnAddedKVProcessor2_0

[[autodoc]] models.attention_processor.AttnAddedKVProcessor2_0

CrossFrameAttnProcessor

[[autodoc]] pipelines.text_to_video_synthesis.pipeline_text_to_video_zero.CrossFrameAttnProcessor

CustomDiffusionAttnProcessor

[[autodoc]] models.attention_processor.CustomDiffusionAttnProcessor

CustomDiffusionAttnProcessor2_0

[[autodoc]] models.attention_processor.CustomDiffusionAttnProcessor2_0

CustomDiffusionXFormersAttnProcessor

[[autodoc]] models.attention_processor.CustomDiffusionXFormersAttnProcessor

FusedAttnProcessor2_0

[[autodoc]] models.attention_processor.FusedAttnProcessor2_0

SlicedAttnProcessor

[[autodoc]] models.attention_processor.SlicedAttnProcessor

SlicedAttnAddedKVProcessor

[[autodoc]] models.attention_processor.SlicedAttnAddedKVProcessor

XFormersAttnProcessor

[[autodoc]] models.attention_processor.XFormersAttnProcessor

AttnProcessorNPU

[[autodoc]] models.attention_processor.AttnProcessorNPU