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#
Functions#
Data#
API#
- nemo_automodel.components.models.gemma4_drafter.model._make_missing(name: str)#
- nemo_automodel.components.models.gemma4_drafter.model.__all__#
[‘Gemma4DrafterForCausalLM’]