Skip to content

Commit 505afa9

Browse files
authored
Add Gemma 3 multimodal SFT test to multipod post-training DAG (GoogleCloudPlatform#1298)
1 parent d58657c commit 505afa9

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

dags/multipod/maxtext_e2e_tpu_post_training.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,10 @@ def get_workload_name(model, mode, length=6):
6363
"command": "bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_sft.sh",
6464
"maxtext_ckpt_path": "gs://runner-maxtext-logs/gemma3-4b/sft/{run_name}/checkpoints/5/model_params",
6565
},
66+
"multimodal_sft": {
67+
"command": "bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_multimodal_sft.sh",
68+
"maxtext_ckpt_path": "gs://runner-maxtext-logs/gemma3-4b/multimodal_sft/{run_name}/checkpoints/5/model_params",
69+
},
6670
"rl": {
6771
"command": "bash tests/end_to_end/tpu/gemma3/4b/test_gemma3_rl.sh",
6872
"maxtext_ckpt_path": "gs://runner-maxtext-logs/gemma3-4b/rl/{run_name}/checkpoints/actor/5/model_params",

0 commit comments

Comments
 (0)