Fix ladder bug

allenai · Aug 31, 2024 · 6d6eeb9 · 6d6eeb9
1 parent de8721b
commit 6d6eeb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/olmo/scaling/ladder.py b/olmo/scaling/ladder.py
@@ -525,7 +525,7 @@ def train_cmd(args: argparse.Namespace):
     prepare_cli_environment()
     add_cached_path_clients()
 
-    from train import main
+    from olmo.train import main
 
     main(cfg)