Request for new models

[karthik4579]

Will support for new models be added like Auraflow and Flux be added ?

https://huggingface.co/fal/AuraFlow
https://huggingface.co/black-forest-labs/FLUX.1-dev

[entrpn]

@karthik4579 there are plans to add Flux in the future

[gabeweisz]

Can you comment on the expected timeline for adding flux? Would you be open to a PR that adds it (we haven't built it yet, but could work on it if you are open to such a PR)

[entrpn]

@gabeweisz PRs are always welcomed :)

I am planning to start working on flux in beginning of 2025. There is already a JAX Flux repo that runs on both GPUs and TPUs, but it needs to be optimized. repo