cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout#

class cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout(*args, **kwargs)[source]#

Bases: torch.nn.Module

Methods

`forward`
`token_drop`

previous

cerebras.modelzoo.data.vision.diffusion.dit_transforms

next

cerebras.modelzoo.data.vision.diffusion.dit_transforms.NoiseGenerator