bridge.diffusion.models

Subpackages

• bridge.diffusion.models.common
  • bridge.diffusion.models.common.normalization
  • bridge.diffusion.models.common.dit_embeddings
  • bridge.diffusion.models.common.nemotron_labs_diffusion_attention
  • bridge.diffusion.models.common.dit_attention
  • bridge.diffusion.models.common.dgpt_step
• bridge.diffusion.models.nemotron_labs_diffusion
  • bridge.diffusion.models.nemotron_labs_diffusion.nemotron_labs_diffusion_provider
  • bridge.diffusion.models.nemotron_labs_diffusion.inference_nemotron_labs_diffusion
• bridge.diffusion.models.flux
  • bridge.diffusion.models.flux.flow_matching
    • bridge.diffusion.models.flux.flow_matching.flux_inference_pipeline
    • bridge.diffusion.models.flux.flow_matching.flux_adapter
  • bridge.diffusion.models.flux.layers
  • bridge.diffusion.models.flux.flux_layer_spec
  • bridge.diffusion.models.flux.flux_attention
  • bridge.diffusion.models.flux.flux_model
  • bridge.diffusion.models.flux.flux_step
  • bridge.diffusion.models.flux.flux_provider
• bridge.diffusion.models.wan
  • bridge.diffusion.models.wan.inference
    • bridge.diffusion.models.wan.inference.utils
  • bridge.diffusion.models.wan.flow_matching
    • bridge.diffusion.models.wan.flow_matching.flow_matching_pipeline_wan
    • bridge.diffusion.models.wan.flow_matching.flow_inference_pipeline
  • bridge.diffusion.models.wan.wan_provider
  • bridge.diffusion.models.wan.utils
  • bridge.diffusion.models.wan.wan_layer_spec
  • bridge.diffusion.models.wan.wan_model
  • bridge.diffusion.models.wan.rope_utils
  • bridge.diffusion.models.wan.wan_step

Package Contents

Data

__all__

API

bridge.diffusion.models.__all__

[‘NemotronLabsDiffusionBridge’]