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 = Gemma4DrafterForCausalLM

nemo_automodel.components.models.gemma4_drafter.__all__ = ['Gemma4DrafterForCausalLM', 'Gemma4JointOutput', 'Gemma4WithDrafter']