> This page is for version 0.5.0 · 26.06.
> For other versions, use one of these documentation indexes:
> - Nightly (default): https://docs.nvidia.com/nemo/automodel/nightly/llms.txt
> - Latest: https://docs.nvidia.com/nemo/automodel/latest/llms.txt
> - 0.5.0 · 26.06: https://docs.nvidia.com/nemo/automodel/v0.5/llms.txt
> - 0.4.0 · 26.04: https://docs.nvidia.com/nemo/automodel/v0.4/llms.txt

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

# nemo_automodel.components.datasets.diffusion.loader

Typed result and construction contract for diffusion dataloaders.

## Module Contents

### Classes

| Name                                                                                                          | Description                                                                  |
| ------------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------- |
| [`DiffusionDataloaderBuild`](#nemo_automodel-components-datasets-diffusion-loader-DiffusionDataloaderBuild)   | Materialized diffusion dataloader, sampler, and effective batch-tail policy. |
| [`DiffusionDataloaderConfig`](#nemo_automodel-components-datasets-diffusion-loader-DiffusionDataloaderConfig) | Typed construction contract used by the diffusion recipe.                    |

### API

```python
class nemo_automodel.components.datasets.diffusion.loader.DiffusionDataloaderBuild(
    dataloader: torch.utils.data.DataLoader,
    sampler: torch.utils.data.Sampler[object] | None,
    drop_last: bool = False
)
```

Dataclass

Materialized diffusion dataloader, sampler, and effective batch-tail policy.

**`dataloader`** `DataLoader`

---

**`drop_last`** `bool = False`

---

**`sampler`** `Sampler[object] | None`

---

```python
class nemo_automodel.components.datasets.diffusion.loader.DiffusionDataloaderConfig()
```

Protocol

Typed construction contract used by the diffusion recipe.

```python
nemo_automodel.components.datasets.diffusion.loader.DiffusionDataloaderConfig.build(
    dp_rank: int,
    dp_world_size: int,
    batch_size: int
) -> nemo_automodel.components.datasets.diffusion.loader.DiffusionDataloaderBuild
```

Build the configured per-rank diffusion dataloader.