bridge.diffusion

Subpackages

• bridge.diffusion.common
  • bridge.diffusion.common.flow_matching
    • bridge.diffusion.common.flow_matching.adapters
    • bridge.diffusion.common.flow_matching.flow_matching_pipeline
• bridge.diffusion.conversion
  • bridge.diffusion.conversion.wan
    • bridge.diffusion.conversion.wan.wan_bridge
    • bridge.diffusion.conversion.wan.wan_hf_pretrained
  • bridge.diffusion.conversion.flux
    • bridge.diffusion.conversion.flux.flux_bridge
    • bridge.diffusion.conversion.flux.flux_hf_pretrained
• bridge.diffusion.recipes
  • bridge.diffusion.recipes.wan
    • bridge.diffusion.recipes.wan.wan
  • bridge.diffusion.recipes.flux
    • bridge.diffusion.recipes.flux.flux
• bridge.diffusion.data
  • bridge.diffusion.data.common
    • bridge.diffusion.data.common.sequence_packing_utils
    • bridge.diffusion.data.common.diffusion_sample
    • bridge.diffusion.data.common.diffusion_energon_datamodule
    • bridge.diffusion.data.common.diffusion_task_encoder_with_sp
  • bridge.diffusion.data.wan
    • bridge.diffusion.data.wan.wan_energon_datamodule
    • bridge.diffusion.data.wan.wan_mock_datamodule
    • bridge.diffusion.data.wan.wan_taskencoder
  • bridge.diffusion.data.flux
    • bridge.diffusion.data.flux.flux_energon_datamodule
    • bridge.diffusion.data.flux.flux_taskencoder
    • bridge.diffusion.data.flux.flux_mock_datamodule
• bridge.diffusion.base
• bridge.diffusion.models
  • 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.inference
    • 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.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