nemo_automodel.components.flow_matching

Flow Matching components for diffusion model training.

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

Subpackages

• nemo_automodel.components.flow_matching.adapters
  • nemo_automodel.components.flow_matching.adapters.flux
  • nemo_automodel.components.flow_matching.adapters.base
  • nemo_automodel.components.flow_matching.adapters.hunyuan
  • nemo_automodel.components.flow_matching.adapters.simple

Submodules

• nemo_automodel.components.flow_matching.time_shift_utils
• nemo_automodel.components.flow_matching.pipeline

Package Contents

Functions

__getattr__
__dir__

Data

_LAZY_ATTRS
__all__

API

nemo_automodel.components.flow_matching._LAZY_ATTRS

None

nemo_automodel.components.flow_matching.__all__

‘sorted(…)’

nemo_automodel.components.flow_matching.__getattr__(name: str)

nemo_automodel.components.flow_matching.__dir__()