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),