Implement `CustomDiffusionAttnProcessor2_0`. #4604
Implement `CustomDiffusionAttnProcessor2_0`
3f832508
Doc-strings and type annotations for `CustomDiffusionAttnProcessor2_0…
f0114676
Interops for `CustomDiffusionAttnProcessor2_0`.
e849af4e
eliphatfs
marked this pull request as ready for review 2 years ago
Formatted `attention_processor.py`.
d6889793
Formatted doc-string in `attention_processor.py`
c1678ec8
Conditional CustomDiffusion2_0 for training example.
f6636ada
Remove unnecessary reference impl in comments.
fc693fc6
Fix `save_attn_procs`.
19edb3fa
sayakpaul
approved these changes
on 2023-09-15
Merge branch 'main' into main
f7fc174a
sayakpaul
approved these changes
on 2023-09-18
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub