Fix ladder path

allenai · Aug 31, 2024 · de8721b · de8721b
1 parent 4b01f8a
commit de8721b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/scripts/beaker/ladder-launch.sh b/scripts/beaker/ladder-launch.sh
@@ -35,7 +35,6 @@ gantry run \
   --shared-memory 10GiB \
   --venv base \
   --yes \
-  --timeout=-1 \
   -- /bin/bash -c "${COMMAND}"
 
   # --cluster ai2/pluto-cirrascale \

diff --git a/scripts/beaker/ladder.sh b/scripts/beaker/ladder.sh
@@ -25,4 +25,4 @@ torchrun \
   --rdzv_endpoint=$BEAKER_LEADER_REPLICA_HOSTNAME:29400 \
   --node_rank=$BEAKER_REPLICA_RANK \
   --rdzv_conf="read_timeout=420" \
-  scripts/ladder.py train "$@"
+  olmo/scaling/ladder.py train "$@"