diff --git a/scripts/beaker/ladder-launch.sh b/scripts/beaker/ladder-launch.sh index cc8dcbc64..8cf53996d 100755 --- 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 index c47d23d5d..83cc2a62c 100755 --- 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 "$@"