mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Ruoxi 16b9a57d29 Implement CustomDiffusionAttnProcessor2_0. (#4604 )

* Implement `CustomDiffusionAttnProcessor2_0`

* Doc-strings and type annotations for `CustomDiffusionAttnProcessor2_0`. (#1)

* Update attnprocessor.md

* Update attention_processor.py

* Interops for `CustomDiffusionAttnProcessor2_0`.

* Formatted `attention_processor.py`.

* Formatted doc-string in `attention_processor.py`

* Conditional CustomDiffusion2_0 for training example.

* Remove unnecessary reference impl in comments.

* Fix `save_attn_procs`.

2023-09-18 14:49:00 +02:00

1.3 KiB

Raw Blame History

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

LoRAAttnProcessor

autodoc models.attention_processor.LoRAAttnProcessor

LoRAAttnProcessor2_0

autodoc models.attention_processor.LoRAAttnProcessor2_0

CustomDiffusionAttnProcessor

autodoc models.attention_processor.CustomDiffusionAttnProcessor

CustomDiffusionAttnProcessor2_0

autodoc models.attention_processor.CustomDiffusionAttnProcessor2_0

AttnAddedKVProcessor

autodoc models.attention_processor.AttnAddedKVProcessor

AttnAddedKVProcessor2_0

autodoc models.attention_processor.AttnAddedKVProcessor2_0

LoRAAttnAddedKVProcessor

autodoc models.attention_processor.LoRAAttnAddedKVProcessor

XFormersAttnProcessor

autodoc models.attention_processor.XFormersAttnProcessor

LoRAXFormersAttnProcessor

autodoc models.attention_processor.LoRAXFormersAttnProcessor

CustomDiffusionXFormersAttnProcessor

autodoc models.attention_processor.CustomDiffusionXFormersAttnProcessor

SlicedAttnProcessor

autodoc models.attention_processor.SlicedAttnProcessor

SlicedAttnAddedKVProcessor

autodoc models.attention_processor.SlicedAttnAddedKVProcessor

1.3 KiB Raw Blame History