nemo_automodel.components.models.gemma4_unified.model

NeMo wrapper for the Hugging Face Gemma4 Unified model.

Module Contents

Classes

Name	Description
`Gemma4UnifiedForConditionalGeneration`	Gemma4 Unified with NeMo checkpoint key conversion.

API

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

Bases: HFCheckpointingMixin, HFGemma4UnifiedForConditionalGeneration

Gemma4 Unified with NeMo checkpoint key conversion.

state_dict_adapter

= Gemma4UnifiedStateDictAdapter()

tie_word_embeddings_support

TieSupport = TieSupport.TIED_ONLY

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.