nemo_automodel.components.models.gemma4_drafter.model

Thin NeMo wrapper around HuggingFace Gemma4AssistantForCausalLM.

The HF implementation in transformers.models.gemma4_assistant is used as-is; this wrapper only adds :class:HFCheckpointingMixin so the drafter participates in NeMo’s distributed checkpointing pipeline and gives us a stable native class name for the model registry.

Module Contents

Classes

Name	Description
`Gemma4DrafterForCausalLM`	Placeholder raised when `transformers.models.gemma4_assistant` is unavailable.

Functions

Name	Description
`_make_missing`	-

Data

ModelClass

_GEMMA4_ASSISTANT_HF_AVAILABLE

__all__

API

class nemo_automodel.components.models.gemma4_drafter.model.Gemma4DrafterForCausalLM(
    args = (),
    kwargs = {}
)

Placeholder raised when transformers.models.gemma4_assistant is unavailable.

nemo_automodel.components.models.gemma4_drafter.model.Gemma4DrafterForCausalLM.from_pretrained(
    args = (),
    kwargs = {}
)

classmethod

nemo_automodel.components.models.gemma4_drafter.model._make_missing(
    name: str
)

nemo_automodel.components.models.gemma4_drafter.model.ModelClass = Gemma4DrafterForCausalLM

nemo_automodel.components.models.gemma4_drafter.model._GEMMA4_ASSISTANT_HF_AVAILABLE = True

nemo_automodel.components.models.gemma4_drafter.model.__all__ = ['Gemma4DrafterForCausalLM']