gemma-4-E2B-it | NVIDIA NeMo AutoModel

Gemma-4-E2B-it has a checked-in NeMo AutoModel recipe for image-text-to-text. The Hugging Face configuration declares the Gemma4ForConditionalGeneration architecture.

Fine-Tune Gemma-4-E2B-it

Follow the installation instructions, then review the checked-in recipe and its declared topology below before choosing a local or cluster launcher.

uv run automodel examples/vlm_finetune/gemma4/gemma4_2b.yaml \
  --nproc-per-node 2

Choose a Workflow

Goal	Start Here
Run the primary recipe for this model	Use the recipe configuration.
Try another checked-in recipe for this model	Use the alternate recipe.

Configuration

Setting	Configuration
Hardware	-
Strategy	FSDP2; tp_size=1, cp_size=1
Nodes	1
Features	Checkpointing (enabled=true)
Advanced	torch.bfloat16

More Recipes for This Model

gemma4_2b_peft.yaml

Model Reference

Model Architecture

Property	Value
Task	Image-text-to-text
Architecture	`Gemma4ForConditionalGeneration`

Available Models

google/gemma-4-E2B-it

Vision-language model fine-tuning guide