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

Module Contents#

Classes#

WanDataModuleConfig

API#

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: torch.int_repr#: None

dataloader_type: str#: ‘external’

__post_init__()#

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