nemo_automodel.components.datasets.diffusion

Diffusion model datasets and dataloaders.

Submodules

• nemo_automodel.components.datasets.diffusion.multi_tier_bucketing
• nemo_automodel.components.datasets.diffusion.collate_fns
• nemo_automodel.components.datasets.diffusion.text_to_image_dataset
• nemo_automodel.components.datasets.diffusion.sampler
• nemo_automodel.components.datasets.diffusion.meta_files_dataset
• nemo_automodel.components.datasets.diffusion.mock_dataloader

Package Contents

Functions

__getattr__
__dir__

Data

_LAZY_ATTRS
__all__

API

nemo_automodel.components.datasets.diffusion._LAZY_ATTRS

None

nemo_automodel.components.datasets.diffusion.__all__

‘sorted(…)’

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

nemo_automodel.components.datasets.diffusion.__dir__()