ReferenceFull Library ReferenceNemo AutomodelNemo AutomodelComponentsFlow Matching

nemo_automodel.components.flow_matching.adapters

View as Markdown

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
  • QwenImageAdapter: For Qwen-Image text-to-image models

Submodules

Package Contents

Data

__all__

API

nemo_automodel.components.flow_matching.adapters.__all__ = ['FlowMatchingContext', 'ModelAdapter', 'FluxAdapter', 'Flux2Adapter', 'HunyuanA...