Skip to content

GitLab

Unverified Commit 345907f3 authored by ちくわぶ's avatar ちくわぶ Committed by GitHub
Browse files

Add all AttnProcessor classes in `AttentionProcessor` type (#9909) 

Add all AttnProcessor in `AttentionProcessor` type
parent 07d0fbf3
Hide whitespace changes
Inline Side-by-side
Showing with 36 additions and 9 deletions
src/diffusers/models/attention_processor.py
View file @ 345907f3
...@@ -5053,19 +5053,46 @@ CROSS_ATTENTION_PROCESSORS = ( ...@@ -5053,19 +5053,46 @@ CROSS_ATTENTION_PROCESSORS = (
AttentionProcessor = Union[ AttentionProcessor = Union[
AttnProcessor, AttnProcessor,
AttnProcessor2_0, CustomDiffusionAttnProcessor,
FusedAttnProcessor2_0,
XFormersAttnProcessor,
SlicedAttnProcessor,
AttnAddedKVProcessor, AttnAddedKVProcessor,
SlicedAttnAddedKVProcessor,
AttnAddedKVProcessor2_0, AttnAddedKVProcessor2_0,
JointAttnProcessor2_0,
PAGJointAttnProcessor2_0,
PAGCFGJointAttnProcessor2_0,
FusedJointAttnProcessor2_0,
AllegroAttnProcessor2_0,
AuraFlowAttnProcessor2_0,
FusedAuraFlowAttnProcessor2_0,
FluxAttnProcessor2_0,
FluxAttnProcessor2_0_NPU,
FusedFluxAttnProcessor2_0,
FusedFluxAttnProcessor2_0_NPU,
CogVideoXAttnProcessor2_0,
FusedCogVideoXAttnProcessor2_0,
XFormersAttnAddedKVProcessor, XFormersAttnAddedKVProcessor,
CustomDiffusionAttnProcessor, XFormersAttnProcessor,
AttnProcessorNPU,
AttnProcessor2_0,
MochiVaeAttnProcessor2_0,
StableAudioAttnProcessor2_0,
HunyuanAttnProcessor2_0,
FusedHunyuanAttnProcessor2_0,
PAGHunyuanAttnProcessor2_0,
PAGCFGHunyuanAttnProcessor2_0,
LuminaAttnProcessor2_0,
MochiAttnProcessor2_0,
FusedAttnProcessor2_0,
CustomDiffusionXFormersAttnProcessor, CustomDiffusionXFormersAttnProcessor,
CustomDiffusionAttnProcessor2_0, CustomDiffusionAttnProcessor2_0,
PAGCFGIdentitySelfAttnProcessor2_0, SlicedAttnProcessor,
SlicedAttnAddedKVProcessor,
IPAdapterAttnProcessor,
IPAdapterAttnProcessor2_0,
IPAdapterXFormersAttnProcessor,
PAGIdentitySelfAttnProcessor2_0, PAGIdentitySelfAttnProcessor2_0,
PAGCFGHunyuanAttnProcessor2_0, PAGCFGIdentitySelfAttnProcessor2_0,
PAGHunyuanAttnProcessor2_0, LoRAAttnProcessor,
LoRAAttnProcessor2_0,
LoRAXFormersAttnProcessor,
LoRAAttnAddedKVProcessor,
] ]
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