Skip to content

GitLab

Commit 18ebd57b authored by William Berman's avatar William Berman Committed by Will Berman
Browse files

add missing AttnProcessor2_0 to AttentionProcessor union

parent b6cc0502
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/models/attention_processor.py
View file @ 18ebd57b
...@@ -684,6 +684,7 @@ class SlicedAttnAddedKVProcessor: ...@@ -684,6 +684,7 @@ class SlicedAttnAddedKVProcessor:
AttentionProcessor = Union[ AttentionProcessor = Union[
AttnProcessor, AttnProcessor,
AttnProcessor2_0,
XFormersAttnProcessor, XFormersAttnProcessor,
SlicedAttnProcessor, SlicedAttnProcessor,
AttnAddedKVProcessor, AttnAddedKVProcessor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment