shieldgemma2

Model type

shieldgemma2

Class

Gemma2CausalLMModel

Task

text-generation

Source

models/gemma.py

Description

Gemma2 causal language model with alternating local/global attention.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")