nemo_automodel.components.models.diffusion_gemma

NeMo Automodel support for the diffusion_gemma block-diffusion model.

The config classes are the released transformers DiffusionGemmaConfig / DiffusionGemmaTextConfig (transformers defines and AutoConfig-registers model_type="diffusion_gemma" natively), re-exported here for convenience. AM owns only the compute (model/layers/adapter).

Submodules

• nemo_automodel.components.models.diffusion_gemma.attention_mask
• nemo_automodel.components.models.diffusion_gemma.fsdp
• nemo_automodel.components.models.diffusion_gemma.layers
• nemo_automodel.components.models.diffusion_gemma.model
• nemo_automodel.components.models.diffusion_gemma.state_dict_adapter

Package Contents

Data

_MSG

__all__

API

nemo_automodel.components.models.diffusion_gemma._MSG = 'transformers.models.diffusion_gemma is not available.'

nemo_automodel.components.models.diffusion_gemma.__all__ = ['DiffusionGemmaConfig', 'DiffusionGemmaTextConfig']