From c0270366a9006c86498f4158b73ea4ff3fc193b7 Mon Sep 17 00:00:00 2001 From: wenxindongwork <161090399+wenxindongwork@users.noreply.github.com> Date: Thu, 19 Dec 2024 12:05:19 -0800 Subject: [PATCH] Fix typo (#515) --- dags/multipod/maxtext_trillium_configs_perf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dags/multipod/maxtext_trillium_configs_perf.py b/dags/multipod/maxtext_trillium_configs_perf.py index d12e3bab..affb1ba4 100644 --- a/dags/multipod/maxtext_trillium_configs_perf.py +++ b/dags/multipod/maxtext_trillium_configs_perf.py @@ -27,7 +27,7 @@ # Run once a day at 4 am UTC (8 pm PST / 9 pm PDT) SCHEDULED_TIME = "0 4 * * *" if composer_env.is_prod_env() else None -MODEL_CONFIGS = ["gpt3_175b", "llama2_70b_4096", "mixtral_8x7b"] +MODEL_CONFIGS = ["gpt3_175b", "llama2_7b_4096", "mixtral_8x7b"] DOCKER_IMAGES = [ (SetupMode.STABLE, DockerImage.MAXTEXT_TPU_JAX_STABLE_STACK), (SetupMode.NIGHTLY, DockerImage.MAXTEXT_TPU_JAX_NIGHTLY),