bridge.recipes.wan.h100.wan

Module Contents

Functions

wan_1_3b_pretrain_8gpu_h100_bf16_config	Return a pre-training configuration for WAN 1.3B model.
wan_14b_pretrain_8gpu_h100_bf16_config	Return a pre-training configuration for WAN 14B model.
wan_1_3b_sft_8gpu_h100_bf16_config	Return a fine-tuning configuration for WAN 1.3B model.
wan_14b_sft_8gpu_h100_bf16_config	Return a fine-tuning configuration for WAN 14B model.
wan_1_3b_text2image_pretrain_1gpu_h100_bf16_config	Return a Wan 1.3B pretraining configuration tuned for text-to-image data.
wan_1_3b_text2video_pretrain_4gpu_h100_bf16_config	Return a Wan 1.3B pretraining configuration tuned for text-to-video data.

Data

__all__

API

bridge.recipes.wan.h100.wan.wan_1_3b_pretrain_8gpu_h100_bf16_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.recipes.wan.h100.wan.wan_14b_pretrain_8gpu_h100_bf16_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.recipes.wan.h100.wan.wan_1_3b_sft_8gpu_h100_bf16_config() → 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().

bridge.recipes.wan.h100.wan.wan_14b_sft_8gpu_h100_bf16_config() → megatron.bridge.training.config.ConfigContainer

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

Uses the same defaults as wan_14b_pretrain_config().

bridge.recipes.wan.h100.wan.wan_1_3b_text2image_pretrain_1gpu_h100_bf16_config() → megatron.bridge.training.config.ConfigContainer

Return a Wan 1.3B pretraining configuration tuned for text-to-image data.

Wraps wan_1_3b_pretrain_config and overrides sequence length on both the model and the dataset for spatial-only inputs.

bridge.recipes.wan.h100.wan.wan_1_3b_text2video_pretrain_4gpu_h100_bf16_config() → megatron.bridge.training.config.ConfigContainer

Return a Wan 1.3B pretraining configuration tuned for text-to-video data.

Wraps wan_1_3b_pretrain_config and overrides sequence length on both the model and the dataset for spatio-temporal inputs, with context parallelism reduced to 4 to fit the longer sequence.

bridge.recipes.wan.h100.wan.__all__

[‘wan_14b_pretrain_8gpu_h100_bf16_config’, ‘wan_14b_sft_8gpu_h100_bf16_config’, ‘wan_1_3b_pretrain_8…