gm.ckpts.CheckpointPath

class gemma.gm.ckpts.CheckpointPath(*values)

Bases: enum.StrEnum

Hardcoded paths to Gemma checkpoints.

Format: {VERSION}_{SIZE}_{VARIANT}.

Variants:

• PT: Pre-trained

• IT: Instruction Tuned (most likely the model you want to use)

For example, GEMMA2_27B_IT is Gemma V2, 27 Billion parameters, instruction tuned.

GEMMA2_2B_PT = 'gs://gemma-data/checkpoints/gemma2-2b-pt'

GEMMA2_9B_PT = 'gs://gemma-data/checkpoints/gemma2-9b-pt'

GEMMA2_27B_PT = 'gs://gemma-data/checkpoints/gemma2-27b-pt'

GEMMA2_2B_IT = 'gs://gemma-data/checkpoints/gemma2-2b-it'

GEMMA2_9B_IT = 'gs://gemma-data/checkpoints/gemma2-9b-it'

GEMMA2_27B_IT = 'gs://gemma-data/checkpoints/gemma2-27b-it'

GEMMA3_270M_PT = 'gs://gemma-data/checkpoints/gemma3-270m-pt'

GEMMA3_1B_PT = 'gs://gemma-data/checkpoints/gemma3-1b-pt'

GEMMA3_4B_PT = 'gs://gemma-data/checkpoints/gemma3-4b-pt'

GEMMA3_12B_PT = 'gs://gemma-data/checkpoints/gemma3-12b-pt'

GEMMA3_27B_PT = 'gs://gemma-data/checkpoints/gemma3-27b-pt'

GEMMA3_270M_IT = 'gs://gemma-data/checkpoints/gemma3-270m-it'

GEMMA3_1B_IT = 'gs://gemma-data/checkpoints/gemma3-1b-it'

GEMMA3_4B_IT = 'gs://gemma-data/checkpoints/gemma3-4b-it'

GEMMA3_12B_IT = 'gs://gemma-data/checkpoints/gemma3-12b-it'

GEMMA3_27B_IT = 'gs://gemma-data/checkpoints/gemma3-27b-it'

GEMMA3N_E2B_PT = 'gs://gemma-data/checkpoints/gemma3n-e2b-pt'

GEMMA3N_E4B_PT = 'gs://gemma-data/checkpoints/gemma3n-e4b-pt'

GEMMA3N_E2B_IT = 'gs://gemma-data/checkpoints/gemma3n-e2b-it'

GEMMA3N_E4B_IT = 'gs://gemma-data/checkpoints/gemma3n-e4b-it'

GEMMA4_E2B_PT = 'gs://gemma-data/checkpoints/gemma4-e2b-pt'

GEMMA4_E4B_PT = 'gs://gemma-data/checkpoints/gemma4-e4b-pt'

GEMMA4_26B_A4B_PT = 'gs://gemma-data/checkpoints/gemma4-26b-a4b-pt'

GEMMA4_31B_PT = 'gs://gemma-data/checkpoints/gemma4-31b-pt'

GEMMA4_E2B_IT = 'gs://gemma-data/checkpoints/gemma4-e2b-it'

GEMMA4_E4B_IT = 'gs://gemma-data/checkpoints/gemma4-e4b-it'

GEMMA4_26B_A4B_IT = 'gs://gemma-data/checkpoints/gemma4-26b-a4b-it'

GEMMA4_31B_IT = 'gs://gemma-data/checkpoints/gemma4-31b-it'