PipeFusion Support for FLUX-2

[ZRY000]

Hi, amazing framework — thanks for the great work!
I noticed that in the FLUX-2 example script examples/flux2_example.py, PipeFusion is currently not supported:

assert engine_args.pipefusion_parallel_degree == 1, "This script does not support PipeFusion."

https://github.com/xdit-project/xDiT/blob/main/examples/flux2_example.py#L103

Do you have a plan or timeline to support PipeFusion for FLUX-2?
I’m very interested in using PipeFusion for FLUX-2 since FLUX-2 is quite large.

Looking forward to future updates and new commits — thanks again!

[avjves]

Hi!

In the immediate future we don't have plans to implement pipefusion for FLUX.2. Though we are interested in looking into this in the future.

Any PRs to add said support would also be very welcome if you feel like giving it a shot :)