> 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.1-dev

> Use FLUX.1-dev with NeMo AutoModel for diffusion training and fine-tuning, with documented checkpoints, runnable recipes, setup guidance, and model reference details.

[FLUX.1-dev](https://huggingface.co/black-forest-labs/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](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/nemo-automodel) or follow the [installation instructions](/get-started/installation).

## Fine-Tune FLUX.1-dev

From the repository root, run:

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

## Choose a Workflow

| Goal                                      | Start Here                                                                                                                     |
| ----------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------ |
| Fine-tune - FLUX.1-dev with flow matching | Use [flux\_t2i\_flow.yaml](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/diffusion/finetune/flux_t2i_flow.yaml). |
| Pretrain - FLUX.1-dev with flow matching  | Use [flux\_t2i\_flow.yaml](https://github.com/NVIDIA-NeMo/Automodel/blob/main/examples/diffusion/pretrain/flux_t2i_flow.yaml). |

## Model Reference

### Model Architecture

| Property                  | Value                                                         |
| ------------------------- | ------------------------------------------------------------- |
| Task                      | Text-to-Image                                                 |
| Architecture              | DiT (Flow Matching)                                           |
| Parameters                | 12B                                                           |
| Hugging Face Organization | [black-forest-labs](https://huggingface.co/black-forest-labs) |

### Task

* Text-to-Image (T2I)

### Available Models

| Model      | HF ID                                                                                 |
| ---------- | ------------------------------------------------------------------------------------- |
| FLUX.1-dev | [`black-forest-labs/FLUX.1-dev`](https://huggingface.co/black-forest-labs/FLUX.1-dev) |

## Related Resources

* [Diffusion Fine-Tuning Guide](/recipes-e2e-examples/diffusion-fine-tuning)
* [Dataset Preparation](/datasets/diffusion-dataset)