Ad

flux2: Open-weight inference for FLUX.2 models

Deploy FLUX.2 models for real-time image generation and editing. This repository provides minimal inference code for various FLUX.2 models, including distilled and base versions, for both text-to-image and image editing tasks.
Screenshot of black-forest-labs/flux2 homepage

FLUX.2 enables state-of-the-art image generation and editing from Black Forest Labs. This repository provides a lightweight inference framework for running the FLUX.2 open-weight models. It addresses the need for efficient and flexible image generation solutions, particularly emphasizing sub-second generation times and compatibility with consumer GPUs. The core technology utilizes a transformer-based architecture optimized for speed and quality.

This project distinguishes itself through its focus on achieving high quality while maintaining extremely low latency, particularly with the 'klein' family of models. It supports both text-to-image and single/multi-reference image editing, offering unified functionality in a single model. The resource efficiency of the 'klein' models allows deployment on consumer GPUs like the RTX 3090/4070, making it accessible for a wider range of users. The use of Apache 2.0 licensing for the 4B models promotes open-source adoption and customization.

  • Sub-second Inference: Enables real-time image generation and editing on modern hardware.
  • Unified Generation & Editing: Supports text-to-image, single-reference, and multi-reference editing with a single model.
  • Consumer GPU Compatibility: The 'klein' 4B model fits within ~8GB VRAM on RTX 3090/4070 and above.
  • Flexible Licensing: 4B models are Apache 2.0 licensed, and 9B models utilize the FLUX.2-dev Non-Commercial License.
  • Extensible Architecture: Designed for fine-tuning, LoRA training, and customization.
  • Integrated Watermarking: Includes an option to embed invisible watermarks into generated images.
  • Optimized for Performance: Achieves a Pareto frontier for quality versus latency and VRAM.

The project is actively maintained with recent commits and ongoing development. The documentation is relatively comprehensive, providing clear instructions for setup, usage, and advanced features like upsampling and Markdown watermarking. The active community and regular model releases indicate a healthy and evolving project. The presence of detailed benchmark results and performance comparisons reinforces its reliability.

This project benefits researchers, developers, and artists seeking high-quality image generation and editing capabilities with low latency and resource requirements. It's ideal for real-time applications, creative workflows, and experimentation on consumer hardware. It offers a valuable alternative to closed-source APIs or computationally expensive solutions, providing open-source flexibility and control over the image generation process.

Languages:
Summarize:
Share:
Stars
2,483
Forks
175
Issues
37
Created
9 months ago
Commit
5 months ago
License
APACHE-2.0
Archived
No
Updated 28 days ago

Similar Repositories