bridge.diffusion.models.common

Common modules for diffusion models in DFM.

Submodules

• bridge.diffusion.models.common.normalization
• bridge.diffusion.models.common.dit_embeddings
• bridge.diffusion.models.common.nemotron_labs_diffusion_attention
• bridge.diffusion.models.common.dit_attention
• bridge.diffusion.models.common.dgpt_step

Package Contents

Data

__all__

API

bridge.diffusion.models.common.__all__

[‘DiTCrossAttention’, ‘DiTCrossAttentionSubmodules’, ‘DiTSelfAttention’, ‘ParallelTimestepEmbedding’…