@@ -5053,19 +5053,46 @@ def __init__(self):
5053
5053
5054
5054
AttentionProcessor = Union [
5055
5055
AttnProcessor ,
5056
- AttnProcessor2_0 ,
5057
- FusedAttnProcessor2_0 ,
5058
- XFormersAttnProcessor ,
5059
- SlicedAttnProcessor ,
5056
+ CustomDiffusionAttnProcessor ,
5060
5057
AttnAddedKVProcessor ,
5061
- SlicedAttnAddedKVProcessor ,
5062
5058
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 ,
5063
5072
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 ,
5065
5085
CustomDiffusionXFormersAttnProcessor ,
5066
5086
CustomDiffusionAttnProcessor2_0 ,
5067
- PAGCFGIdentitySelfAttnProcessor2_0 ,
5087
+ SlicedAttnProcessor ,
5088
+ SlicedAttnAddedKVProcessor ,
5089
+ IPAdapterAttnProcessor ,
5090
+ IPAdapterAttnProcessor2_0 ,
5091
+ IPAdapterXFormersAttnProcessor ,
5068
5092
PAGIdentitySelfAttnProcessor2_0 ,
5069
- PAGCFGHunyuanAttnProcessor2_0 ,
5070
- PAGHunyuanAttnProcessor2_0 ,
5093
+ PAGCFGIdentitySelfAttnProcessor2_0 ,
5094
+ LoRAAttnProcessor ,
5095
+ LoRAAttnProcessor2_0 ,
5096
+ LoRAXFormersAttnProcessor ,
5097
+ LoRAAttnAddedKVProcessor ,
5071
5098
]
0 commit comments