> 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.gemma4_unified.model

NeMo wrapper for the Hugging Face Gemma4 Unified model.

## Module Contents

### Classes

| Name                                                                                                                                    | Description                                         |
| --------------------------------------------------------------------------------------------------------------------------------------- | --------------------------------------------------- |
| [`Gemma4UnifiedForConditionalGeneration`](#nemo_automodel-components-models-gemma4_unified-model-Gemma4UnifiedForConditionalGeneration) | Gemma4 Unified with NeMo checkpoint key conversion. |

### API

```python
class nemo_automodel.components.models.gemma4_unified.model.Gemma4UnifiedForConditionalGeneration(
    config
)
```

**Bases:** [HFCheckpointingMixin](/nemo-automodel/nemo_automodel/components/models/common/hf_checkpointing_mixin#nemo_automodel-components-models-common-hf_checkpointing_mixin-HFCheckpointingMixin), `HFGemma4UnifiedForConditionalGeneration`

Gemma4 Unified with NeMo checkpoint key conversion.

**`state_dict_adapter`** `= Gemma4UnifiedStateDictAdapter()`

---

**`tie_word_embeddings_support`** `TieSupport = TieSupport.TIED_ONLY`

---

```python
nemo_automodel.components.models.gemma4_unified.model.Gemma4UnifiedForConditionalGeneration.tie_weights(
    _args: object = (),
    _kwargs: object = {}
) -> None
```

Tie the output head to Gemma4 Unified's text input embedding.