Testing on MMDiT based models: SD3 and Flux

[adyk007]

Hi authors,
Thank you for this repo. As far as I can understand this framework will not work for newer T2I Diffusion Models? Would you throw some light on this?