From 447f68c73392d57d9f77c694a83acf02a1746728 Mon Sep 17 00:00:00 2001
From: dwadden <davidw@allenai.org>
Date: Fri, 12 Apr 2024 01:17:26 +0000
Subject: [PATCH] Set the cluster.

---
 scripts/beaker/annealing/launch_annealing.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/beaker/annealing/launch_annealing.sh b/scripts/beaker/annealing/launch_annealing.sh
index 7fdc5a68b..61771dee3 100755
--- a/scripts/beaker/annealing/launch_annealing.sh
+++ b/scripts/beaker/annealing/launch_annealing.sh
@@ -18,7 +18,7 @@ function launch {
     --description ${CONFIG_NAME} \
     --priority high \
     --beaker-image petew/olmo-torch2-gantry \
-    --cluster ai2/pluto-cirrascale \
+    --cluster $CLUSTER \
     --gpus 8 \
     --replicas "${NUM_NODES}" \
     --leader-selection \