diff --git a/docs/Setup-Internal.md b/docs/Setup-Internal.md index ea2381d74..1a5d9b117 100644 --- a/docs/Setup-Internal.md +++ b/docs/Setup-Internal.md @@ -43,7 +43,7 @@ Create a GitHub token that can clone this repo on Beaker. Generate a token [here ```bash beaker config set default_workspace ai2/earth-systems -beaker workspace set-budget ai2/earth-systems ai2/es-platform +beaker workspace set-budget ai2/earth-systems ai2/atec-olmoearth ``` ### 3. Set Beaker Secrets @@ -144,7 +144,7 @@ Evaluation datasets have default paths configured in [`olmoearth_pretrain/evals/ ## Beaker Information **Quick Reference:** -- **Budget:** `ai2/es-platform` +- **Budget:** `ai2/atec-olmoearth` - **Workspace:** `ai2/earth-systems` - **Weka:** `weka://dfive-default` diff --git a/olmoearth_pretrain/dataset_creation/internal_docs.md b/olmoearth_pretrain/dataset_creation/internal_docs.md index ba04b3641..1b9282641 100644 --- a/olmoearth_pretrain/dataset_creation/internal_docs.md +++ b/olmoearth_pretrain/dataset_creation/internal_docs.md @@ -4,7 +4,7 @@ A Beaker session can be used to run most of the window creation and data materialization steps: ``` -beaker session create --budget ai2/es-platform --workspace ai2/earth-systems --priority high --gpus 1 --shared-memory 128GiB --bare --mount src=weka,ref=dfive-default,dst=/weka/dfive-default +beaker session create --budget ai2/atec-olmoearth --workspace ai2/earth-systems --priority high --gpus 1 --shared-memory 128GiB --bare --mount src=weka,ref=dfive-default,dst=/weka/dfive-default ``` The only exception is for Sentinel-1 and Sentinel-2 L2A, where it may be desirable to diff --git a/olmoearth_pretrain/inference_benchmarking/constants.py b/olmoearth_pretrain/inference_benchmarking/constants.py index 6ff33d859..2564776f3 100644 --- a/olmoearth_pretrain/inference_benchmarking/constants.py +++ b/olmoearth_pretrain/inference_benchmarking/constants.py @@ -1,7 +1,7 @@ """Shared constants important for inference throughput benchmarking.""" # BEAKER-LAND -BEAKER_BUDGET = "ai2/es-platform" +BEAKER_BUDGET = "ai2/atec-olmoearth" BEAKER_WORKSPACE = "ai2/earth-systems" WEKA_BUCKET = "dfive-default" BEAKER_TASK_PRIORITY = "normal" diff --git a/olmoearth_pretrain/internal/common.py b/olmoearth_pretrain/internal/common.py index bb136611c..7ba63e07f 100644 --- a/olmoearth_pretrain/internal/common.py +++ b/olmoearth_pretrain/internal/common.py @@ -24,7 +24,7 @@ ) logger = logging.getLogger(__name__) -BUDGET = "ai2/es-platform" +BUDGET = "ai2/atec-olmoearth" WORKSPACE = "ai2/earth-systems" DEFAULT_OLMOEARTH_PRETRAIN_WEKA_BUCKET = BeakerWekaBucket(