nemo_automodel.components.flow_matching

View as Markdown

Flow Matching components for diffusion model training.

Provides FlowMatchingPipeline and model-specific adapters for training diffusion models using the flow matching objective.

Subpackages

Submodules

Package Contents

Functions

NameDescription
__dir__-
__getattr__-

Data

_LAZY_ATTRS

__all__

API

nemo_automodel.components.flow_matching.__dir__()
nemo_automodel.components.flow_matching.__getattr__(
name: str
)
nemo_automodel.components.flow_matching._LAZY_ATTRS = {'FlowMatchingPipeline': ('.pipeline', 'FlowMatchingPipeline'), 'create_adapter'...
nemo_automodel.components.flow_matching.__all__ = sorted(_LAZY_ATTRS.keys())