@@ -5053,19 +5053,46 @@ def __init__(self):
50535053
50545054AttentionProcessor = 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