nemo_automodel.components.flow_matching.adapters

Model adapters for FlowMatching Pipeline.

This module provides model-specific adapters that decouple the flow matching logic from model-specific implementation details.

Available Adapters:

• ModelAdapter: Abstract base class for all adapters

• HunyuanAdapter: For HunyuanVideo 1.5 style models

• SimpleAdapter: For simple transformer models (e.g., Wan)

• FluxAdapter: For FLUX.1 text-to-image models

Usage: from automodel.flow_matching.adapters import HunyuanAdapter, SimpleAdapter, FluxAdapter

# Or import the base class to create custom adapters
from automodel.flow_matching.adapters import ModelAdapter

Submodules

• 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

Package Contents

Data

__all__

API

nemo_automodel.components.flow_matching.adapters.__all__

[‘FlowMatchingContext’, ‘ModelAdapter’, ‘FluxAdapter’, ‘HunyuanAdapter’, ‘SimpleAdapter’]