> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# FLUX.2-dev

> Supported FLUX.2-dev checkpoint and NeMo AutoModel diffusion recipes.

[FLUX.2-dev](https://huggingface.co/black-forest-labs/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](https://huggingface.co/black-forest-labs) |

## Example Model and Recipes

| Model                                                                                 | Recipe                                                                                                                 |
| ------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------------------------------------------------- |
| [`black-forest-labs/FLUX.2-dev`](https://huggingface.co/black-forest-labs/FLUX.2-dev) | [Full fine-tuning](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/diffusion/finetune/flux2_t2i_flow.yaml) |
| [`black-forest-labs/FLUX.2-dev`](https://huggingface.co/black-forest-labs/FLUX.2-dev) | [LoRA](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/diffusion/finetune/flux2_t2i_flow_lora.yaml)        |

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

See the [Diffusion Fine-Tuning Guide](/recipes-e2e-examples/diffusion-fine-tuning).