FLUX.2-dev | NVIDIA NeMo AutoModel

FLUX.2-dev is a text-to-image diffusion model supported by NeMo AutoModel’s flow-matching training recipe.


Task	Text-to-Image
Architecture	Diffusion Transformer (Flow Matching)
HF Org	black-forest-labs

Example Model and Recipes

Model	Recipe
`black-forest-labs/FLUX.2-dev`	Full fine-tuning
`black-forest-labs/FLUX.2-dev`	LoRA

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