> 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 and its sampler.        |
| [`DiffusionDataloaderConfig`](#nemo_automodel-components-datasets-diffusion-loader-DiffusionDataloaderConfig) | Typed construction contract used by the diffusion recipe. |

### Data

[`__all__`](#nemo_automodel-components-datasets-diffusion-loader-__all__)

### API

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

Dataclass

Materialized diffusion dataloader and its sampler.

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

```python
nemo_automodel.components.datasets.diffusion.loader.__all__ = ['DiffusionDataloaderBuild', 'DiffusionDataloaderConfig']
```