`bridge.diffusion.conversion.wan.wan_bridge`#

Module Contents#

Megatron Bridge for WAN model.

class bridge.diffusion.conversion.wan.wan_bridge.WanBridge#

Bases: megatron.bridge.models.conversion.model_bridge.MegatronModelBridge

Megatron Bridge for WAN model.

As a user you would not use this bridge directly, but through AutoBridge.

.. rubric:: Example

from megatron.bridge import AutoBridge bridge = AutoBridge.from_hf_pretrained(“WAN-3D-1.3B-v1”) provider = bridge.to_megatron_provider()

provider_bridge( hf_pretrained: megatron.bridge.diffusion.conversion.wan.wan_hf_pretrained.PreTrainedWAN, ) → megatron.bridge.diffusion.models.wan.wan_provider.WanModelProvider#

mapping_registry() → megatron.bridge.models.conversion.mapping_registry.MegatronMappingRegistry#

Return MegatronMappingRegistry containing parameter mappings from HF to Megatron format.