bridge.diffusion.models

Subpackages

• bridge.diffusion.models.common
  • bridge.diffusion.models.common.normalization
  • bridge.diffusion.models.common.dit_attention
  • bridge.diffusion.models.common.dit_embeddings
• bridge.diffusion.models.wan
  • bridge.diffusion.models.wan.flow_matching
    • bridge.diffusion.models.wan.flow_matching.flow_inference_pipeline
    • bridge.diffusion.models.wan.flow_matching.flow_matching_pipeline_wan
  • bridge.diffusion.models.wan.inference
    • bridge.diffusion.models.wan.inference.utils
  • bridge.diffusion.models.wan.wan_provider
  • bridge.diffusion.models.wan.utils
  • bridge.diffusion.models.wan.wan_model
  • bridge.diffusion.models.wan.wan_layer_spec
  • bridge.diffusion.models.wan.rope_utils
  • bridge.diffusion.models.wan.wan_step
• 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.flux_attention
  • bridge.diffusion.models.flux.flux_provider
  • bridge.diffusion.models.flux.flux_layer_spec
  • bridge.diffusion.models.flux.layers
  • bridge.diffusion.models.flux.flux_step
  • bridge.diffusion.models.flux.flux_model