Diffusers Flux-2 adds Flux2Pipeline support for Mistral3 models; Flax deprecated, OOM risk observed

AI Impact Summary

Diffusers introduces Flux-2 integration via Flux2Pipeline and Flux2Transformer2DModel for a 4-bit Flux model (diffusers/FLUX.2-dev-bnb-4bit) using a Mistral3ForConditionalGeneration text encoder. A full run shows substantial memory pressure and an OutOfMemoryError during prompt encoding and image generation, even with enable_model_cpu_offload, indicating VRAM constraints on typical GPUs. The log also repeats a warning that Flax classes are deprecated and will be removed in Diffusers v1.0.0, with a recommendation to migrate to PyTorch classes. For teams, this means readiness for a migration path and tighter capacity planning when using Flux-2 in production.