diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py index 772aae7fcd2f..ffbf4a0056c6 100644 --- a/src/diffusers/models/attention_processor.py +++ b/src/diffusers/models/attention_processor.py @@ -5053,19 +5053,46 @@ def __init__(self): AttentionProcessor = Union[ AttnProcessor, - AttnProcessor2_0, - FusedAttnProcessor2_0, - XFormersAttnProcessor, - SlicedAttnProcessor, + CustomDiffusionAttnProcessor, AttnAddedKVProcessor, - SlicedAttnAddedKVProcessor, 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, - CustomDiffusionAttnProcessor, + XFormersAttnProcessor, + AttnProcessorNPU, + AttnProcessor2_0, + MochiVaeAttnProcessor2_0, + StableAudioAttnProcessor2_0, + HunyuanAttnProcessor2_0, + FusedHunyuanAttnProcessor2_0, + PAGHunyuanAttnProcessor2_0, + PAGCFGHunyuanAttnProcessor2_0, + LuminaAttnProcessor2_0, + MochiAttnProcessor2_0, + FusedAttnProcessor2_0, CustomDiffusionXFormersAttnProcessor, CustomDiffusionAttnProcessor2_0, - PAGCFGIdentitySelfAttnProcessor2_0, + SlicedAttnProcessor, + SlicedAttnAddedKVProcessor, + IPAdapterAttnProcessor, + IPAdapterAttnProcessor2_0, + IPAdapterXFormersAttnProcessor, PAGIdentitySelfAttnProcessor2_0, - PAGCFGHunyuanAttnProcessor2_0, - PAGHunyuanAttnProcessor2_0, + PAGCFGIdentitySelfAttnProcessor2_0, + LoRAAttnProcessor, + LoRAAttnProcessor2_0, + LoRAXFormersAttnProcessor, + LoRAAttnAddedKVProcessor, ]