Renamed test_gamma.sh to test_gemma.sh (#135)

GoogleCloudPlatform · Feb 21, 2024 · f511cd2 · f511cd2
1 parent ad019d4
commit f511cd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/multipod/maxtext_end_to_end.py b/dags/multipod/maxtext_end_to_end.py
@@ -38,7 +38,7 @@
   test_models = {
       "llama2": ["test_llama2", "llama_finetuning_test"],
       "mistral": ["test_mistral"],
-      "gamma": ["test_gamma"],
+      "gemma": ["test_gemma"],
       "gpt3": ["test_gpt3"],
   }
   test_modes = [SetupMode.STABLE, SetupMode.NIGHTLY]