update gcs bucket.

GoogleCloudPlatform · Jan 13, 2024 · ed37286 · ed37286
1 parent 66ddb9f
commit ed37286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/multipod/configs/maxtext_gce_config.py b/dags/multipod/configs/maxtext_gce_config.py
@@ -53,7 +53,7 @@ def get_maxtext_nightly_config(
           "cd /tmp/maxtext && bash setup.sh MODE=nightly;"
           f' JAX_PLATFORM_NAME=TPU XLA_FLAGS="--xla_dump_to=/tmp/xla_dump/" RUN_NAME="{run_name}" &&'
           " python3 MaxText/train.py MaxText/configs/base.yml run_name=$RUN_NAME"
-          f" base_output_directory={gcs_bucket.BENCHMARK_OUTPUT_DIR}"
+          f" base_output_directory={gcs_bucket.MAXTEXT_OUTPUT_DIR}"
           " dataset_path=gs://max-datasets-rogue dataset_type=synthetic"
           " per_device_batch_size=6 reuse_example_batch=1 global_parameter_scale=1 metrics_file='metrics.txt'"
           " steps=50 enable_checkpointing=false enable_profiler=true gcs_metrics=true;"