nemo_automodel.components.flow_matching
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
Submodules
nemo_automodel.components.flow_matching.pipelinenemo_automodel.components.flow_matching.time_shift_utils