Skip to content

Commit 345907f

Browse files
authored
Add all AttnProcessor classes in AttentionProcessor type (#9909)
Add all AttnProcessor in `AttentionProcessor` type
1 parent 07d0fbf commit 345907f

File tree

1 file changed

+36
-9
lines changed

1 file changed

+36
-9
lines changed

src/diffusers/models/attention_processor.py

Lines changed: 36 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5053,19 +5053,46 @@ def __init__(self):
50535053

50545054
AttentionProcessor = Union[
50555055
AttnProcessor,
5056-
AttnProcessor2_0,
5057-
FusedAttnProcessor2_0,
5058-
XFormersAttnProcessor,
5059-
SlicedAttnProcessor,
5056+
CustomDiffusionAttnProcessor,
50605057
AttnAddedKVProcessor,
5061-
SlicedAttnAddedKVProcessor,
50625058
AttnAddedKVProcessor2_0,
5059+
JointAttnProcessor2_0,
5060+
PAGJointAttnProcessor2_0,
5061+
PAGCFGJointAttnProcessor2_0,
5062+
FusedJointAttnProcessor2_0,
5063+
AllegroAttnProcessor2_0,
5064+
AuraFlowAttnProcessor2_0,
5065+
FusedAuraFlowAttnProcessor2_0,
5066+
FluxAttnProcessor2_0,
5067+
FluxAttnProcessor2_0_NPU,
5068+
FusedFluxAttnProcessor2_0,
5069+
FusedFluxAttnProcessor2_0_NPU,
5070+
CogVideoXAttnProcessor2_0,
5071+
FusedCogVideoXAttnProcessor2_0,
50635072
XFormersAttnAddedKVProcessor,
5064-
CustomDiffusionAttnProcessor,
5073+
XFormersAttnProcessor,
5074+
AttnProcessorNPU,
5075+
AttnProcessor2_0,
5076+
MochiVaeAttnProcessor2_0,
5077+
StableAudioAttnProcessor2_0,
5078+
HunyuanAttnProcessor2_0,
5079+
FusedHunyuanAttnProcessor2_0,
5080+
PAGHunyuanAttnProcessor2_0,
5081+
PAGCFGHunyuanAttnProcessor2_0,
5082+
LuminaAttnProcessor2_0,
5083+
MochiAttnProcessor2_0,
5084+
FusedAttnProcessor2_0,
50655085
CustomDiffusionXFormersAttnProcessor,
50665086
CustomDiffusionAttnProcessor2_0,
5067-
PAGCFGIdentitySelfAttnProcessor2_0,
5087+
SlicedAttnProcessor,
5088+
SlicedAttnAddedKVProcessor,
5089+
IPAdapterAttnProcessor,
5090+
IPAdapterAttnProcessor2_0,
5091+
IPAdapterXFormersAttnProcessor,
50685092
PAGIdentitySelfAttnProcessor2_0,
5069-
PAGCFGHunyuanAttnProcessor2_0,
5070-
PAGHunyuanAttnProcessor2_0,
5093+
PAGCFGIdentitySelfAttnProcessor2_0,
5094+
LoRAAttnProcessor,
5095+
LoRAAttnProcessor2_0,
5096+
LoRAXFormersAttnProcessor,
5097+
LoRAAttnAddedKVProcessor,
50715098
]

0 commit comments

Comments
 (0)