Add check for Flux attention processor

2026-09-18 16:54:33 +02:00 · 2024-08-18 00:35:10 +03:00
parent bce3c7eabf
commit 5c857e8aaf
1 changed files with 1 additions and 1 deletions
@@ -1507,7 +1507,7 @@ def set_diffusers_attention(pipe):
        for module in modules:
            if module.__class__.__name__ in ['SD3Transformer2DModel']:
                module.set_attn_processor(p.JointAttnProcessor2_0())
-            elif module.__class__.__name__ in ['HunyuanDiT2DModel']:
+            elif module.__class__.__name__ in ['HunyuanDiT2DModel', 'FluxTransformer2DModel']:
                pass
            else:
                module.set_attn_processor(attention)