Skip to content

GitLab

"docs/source/en/vscode:/vscode.git/clone" did not exist on "b36cbd4fba877a378bd68bb5900e0bb6d99076a8"
Switch branch/tag
Find file BlameHistoryPermalink
  • Ruoxi's avatar
    Implement `CustomDiffusionAttnProcessor2_0`. (#4604) · 16b9a57d
    Ruoxi authored 
    * 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`.
    16b9a57d
train_custom_diffusion.py 57.3 KB