Add double stream support in FluxTransformerBlock for enhanced attent…#4

Open

zhangtao0408 wants to merge 2 commits into

Commits on Nov 5, 2025

Add double stream support in FluxTransformerBlock for enhanced attention processing. Introduce environment variable configuration and optimize attention mechanism based on stream conditions.
TaoZhang-Work
committed
Refactor FluxAttention projection functions and enhance double stream processing in FluxTransformerBlock. Optimize tensor handling and ensure dtype compatibility for improved performance.
TaoZhang-Work
committed