`bridge.diffusion.data.wan.wan_energon_datamodule`#

Module Contents#

Classes#

`WanDatasetConfig`	Unified WAN dataset config: mock vs real is decided at runtime in build_datasets() based on whether `path` is set (same pattern as FLUX recipe with dataset.path).
`WanDataModuleConfig`

Functions#

_is_empty_path

Return True if path should be treated as empty (use mock data).

Data#

logger

API#

bridge.diffusion.data.wan.wan_energon_datamodule.logger#: ‘getLogger(…)’

bridge.diffusion.data.wan.wan_energon_datamodule._is_empty_path(path: Optional[Union[str, list]]) → bool#: Return True if path should be treated as empty (use mock data).

class bridge.diffusion.data.wan.wan_energon_datamodule.WanDatasetConfig#

Bases: megatron.bridge.data.utils.DatasetProvider

Unified WAN dataset config: mock vs real is decided at runtime in build_datasets() based on whether path is set (same pattern as FLUX recipe with dataset.path).

Use this in the recipe with path=None by default. Override with dataset.path=/path/to/wds to load real data; no separate –mock flag needed.

path: Optional[Union[str, list]]#: None

seq_length: int#: 1024

packing_buffer_size: Optional[int]#: None

micro_batch_size: int#: 1

global_batch_size: int#: 4

num_workers: int#: 16

dataloader_type: str#: ‘external’

F_latents: int#: 24

H_latents: int#: 104

W_latents: int#: 60

patch_spatial: int#: 2

patch_temporal: int#: 1

number_packed_samples: int#: 1

context_seq_len: int#: 512

context_embeddings_dim: int#: 4096

__post_init__()#

build_datasets( context: megatron.bridge.data.utils.DatasetBuildContext, )#

class bridge.diffusion.data.wan.wan_energon_datamodule.WanDataModuleConfig#

Bases: megatron.bridge.diffusion.data.common.diffusion_energon_datamodule.DiffusionDataModuleConfig

path: str#: None

seq_length: int#: None

packing_buffer_size: int#: None

micro_batch_size: int#: None

global_batch_size: int#: None

num_workers: int#: None

dataloader_type: str#: ‘external’

__post_init__()#

build_datasets( context: megatron.bridge.data.utils.DatasetBuildContext, )#

bridge.diffusion.data.wan.wan_energon_datamodule#