bridge.diffusion.recipes.wan.wan

Module Contents

Functions

wan_1_3B_pretrain_config	Return a pre-training configuration for WAN 1.3B model.
wan_14B_pretrain_config	Return a pre-training configuration for WAN 14B model.
wan_1_3B_sft_config	Return a fine-tuning configuration for WAN 1.3B model.
wan_14B_sft_config	Return a fine-tuning configuration for WAN 14B model.

API

bridge.diffusion.recipes.wan.wan.wan_1_3B_pretrain_config() → megatron.bridge.training.config.ConfigContainer

Return a pre-training configuration for WAN 1.3B model.

Default parallelism: TP=1, PP=1, CP=8. Uses mock/synthetic data when dataset.path is not set. To use real data, override via CLI: dataset.path=/path/to/wds

bridge.diffusion.recipes.wan.wan.wan_14B_pretrain_config() → megatron.bridge.training.config.ConfigContainer

Return a pre-training configuration for WAN 14B model.

Default parallelism: TP=2, PP=1, CP=4, SP=True. Uses mock/synthetic data when dataset.path is not set. To use real data, override via CLI: dataset.path=/path/to/wds

bridge.diffusion.recipes.wan.wan.wan_1_3B_sft_config(

pretrained_checkpoint: str | None = None,

) → megatron.bridge.training.config.ConfigContainer

Return a fine-tuning configuration for WAN 1.3B model.

Uses the same defaults as wan_1_3B_pretrain_config() and overrides checkpoint to load from pretrained_checkpoint when provided.

bridge.diffusion.recipes.wan.wan.wan_14B_sft_config(

pretrained_checkpoint: str | None = None,

) → megatron.bridge.training.config.ConfigContainer

Return a fine-tuning configuration for WAN 14B model.

Uses the same defaults as wan_14B_pretrain_config() and overrides checkpoint to load from pretrained_checkpoint when provided.