Move ladder.py back to scripts/

allenai · Aug 31, 2024 · 1bff5f8 · 1bff5f8
1 parent 6d6eeb9
commit 1bff5f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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" \
-  olmo/scaling/ladder.py train "$@"
+  scripts/ladder.py train "$@"
diff --git a/olmo/scaling/ladder.py → scripts/ladder.py b/olmo/scaling/ladder.py → scripts/ladder.py
@@ -525,7 +525,7 @@ def train_cmd(args: argparse.Namespace):
     prepare_cli_environment()
     add_cached_path_clients()
 
-    from olmo.train import main
+    from train import main
 
     main(cfg)