FLUX.1-dev

View as Markdown

FLUX.1-dev is a 12B parameter text-to-image diffusion transformer from Black Forest Labs, trained with flow matching. It produces high-fidelity images and is designed for non-commercial research and development use.

Set up NeMo AutoModel with the latest container or follow the installation instructions.

Fine-Tune FLUX.1-dev

From the repository root, run:

uv run torchrun --nproc-per-node=8 \
examples/diffusion/finetune/finetune.py \
-c examples/diffusion/finetune/flux_t2i_flow.yaml

Choose a Workflow

GoalStart Here
Fine-tune - FLUX.1-dev with flow matchingUse flux_t2i_flow.yaml.
Pretrain - FLUX.1-dev with flow matchingUse flux_t2i_flow.yaml.

Model Reference

Model Architecture

PropertyValue
TaskText-to-Image
ArchitectureDiT (Flow Matching)
Parameters12B
Hugging Face Organizationblack-forest-labs

Task

  • Text-to-Image (T2I)

Available Models

ModelHF ID
FLUX.1-devblack-forest-labs/FLUX.1-dev