Can OMG method be transferred to flux model ?

[Adenialzz]

Hi, thanks for open-sourcing.

I wonder can OMG method be implemented to flux model ? Will the difference between MM-DiT attention and UNet attention prevent us transferring the method to flux? I am interested in doing that, hope you can provide some advice. Thanks in advance.

[Adenialzz]

can the spatial relation be preserved in mm-dit ?

[jadechip]

+1