> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# 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/nemo_automodel/components/models/diffusion_gemma/attention_mask)**
* **[`nemo_automodel.components.models.diffusion_gemma.fsdp`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/fsdp)**
* **[`nemo_automodel.components.models.diffusion_gemma.layers`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/layers)**
* **[`nemo_automodel.components.models.diffusion_gemma.model`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/model)**
* **[`nemo_automodel.components.models.diffusion_gemma.state_dict_adapter`](/nemo-automodel/nemo_automodel/components/models/diffusion_gemma/state_dict_adapter)**

## Package Contents

### Data

[`_MSG`](#nemo_automodel-components-models-diffusion_gemma-_MSG)

[`__all__`](#nemo_automodel-components-models-diffusion_gemma-__all__)

### API

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

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