Gantry stuff

jakep-allenai · jakep-allenai · commit dee3730231c7 · 2025-06-26T18:34:53.000Z
diff --git a/gantry-train-requirements.txt b/gantry-train-requirements.txt
@@ -0,0 +1,5 @@
+accelerate
+peft
+wandb
+omegaconf
+einops
diff --git a/scripts/train/newtrainer-gantry.sh b/scripts/train/newtrainer-gantry.sh
@@ -50,7 +50,7 @@ gantry run \
     --beaker-image $BEAKER_USER/$IMAGE_TAG \
     --pip gantry-train-requirements.txt \
     --priority normal \
-    --gpus 8 \
+    --gpus 1 \
     --preemptible \
     --cluster "ai2/jupiter-cirrascale-2" \
     --budget ai2/oe-data \

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +accelerate
 +peft
 +wandb
 +omegaconf
 +einops