nemo_automodel.components.models.gemma4_drafter

Gemma4 drafter (assistant) NeMo Automodel support.

Adds a thin NeMo wrapper around the HuggingFace Gemma4AssistantForCausalLM plus a Gemma4WithDrafter composite for joint base + drafter fine-tuning.

Submodules

• nemo_automodel.components.models.gemma4_drafter.composite
• nemo_automodel.components.models.gemma4_drafter.model

Package Contents

Data

ModelClass
__all__

API

nemo_automodel.components.models.gemma4_drafter.ModelClass

None

nemo_automodel.components.models.gemma4_drafter.__all__

[‘Gemma4DrafterForCausalLM’, ‘Gemma4JointOutput’, ‘Gemma4WithDrafter’]