> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/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

Diffusion model datasets and dataloaders.

## Submodules

* **[`nemo_automodel.components.datasets.diffusion.base_dataset`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/base_dataset)**
* **[`nemo_automodel.components.datasets.diffusion.collate_fns`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/collate_fns)**
* **[`nemo_automodel.components.datasets.diffusion.image_edit_dataset`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/image_edit_dataset)**
* **[`nemo_automodel.components.datasets.diffusion.loader`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/loader)**
* **[`nemo_automodel.components.datasets.diffusion.meta_files_dataset`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/meta_files_dataset)**
* **[`nemo_automodel.components.datasets.diffusion.mock_dataloader`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/mock_dataloader)**
* **[`nemo_automodel.components.datasets.diffusion.multi_tier_bucketing`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/multi_tier_bucketing)**
* **[`nemo_automodel.components.datasets.diffusion.sampler`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/sampler)**
* **[`nemo_automodel.components.datasets.diffusion.text_to_image_dataset`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/text_to_image_dataset)**
* **[`nemo_automodel.components.datasets.diffusion.text_to_video_dataset`](/nemo-automodel/nemo_automodel/components/datasets/diffusion/text_to_video_dataset)**

## Package Contents

### Functions

| Name                                                                       | Description |
| -------------------------------------------------------------------------- | ----------- |
| [`__dir__`](#nemo_automodel-components-datasets-diffusion-__dir__)         | -           |
| [`__getattr__`](#nemo_automodel-components-datasets-diffusion-__getattr__) | -           |

### Data

[`_LAZY_ATTRS`](#nemo_automodel-components-datasets-diffusion-_LAZY_ATTRS)

### API

```python
nemo_automodel.components.datasets.diffusion.__dir__()
```

```python
nemo_automodel.components.datasets.diffusion.__getattr__(
    name: str
)
```

```python
nemo_automodel.components.datasets.diffusion._LAZY_ATTRS = {'BaseMultiresolutionDataset': ('.base_dataset', 'BaseMultiresolutionDataset'), ...
```

## Docs

- [nemo_automodel.components.datasets.diffusion.base_dataset](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/base_dataset.md)
- [nemo_automodel.components.datasets.diffusion.collate_fns](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/collate_fns.md)
- [nemo_automodel.components.datasets.diffusion.image_edit_dataset](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/image_edit_dataset.md)
- [nemo_automodel.components.datasets.diffusion.loader](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/loader.md)
- [nemo_automodel.components.datasets.diffusion.meta_files_dataset](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/meta_files_dataset.md)
- [nemo_automodel.components.datasets.diffusion.mock_dataloader](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/mock_dataloader.md)
- [nemo_automodel.components.datasets.diffusion.multi_tier_bucketing](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/multi_tier_bucketing.md)
- [nemo_automodel.components.datasets.diffusion.sampler](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/sampler.md)
- [nemo_automodel.components.datasets.diffusion.text_to_image_dataset](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/text_to_image_dataset.md)
- [nemo_automodel.components.datasets.diffusion.text_to_video_dataset](https://docs.nvidia.com/nemo/automodel/v0.4/nemo-automodel/nemo_automodel/components/datasets/diffusion/text_to_video_dataset.md)