fix

allenai · Jan 28, 2025 · f3a06d2 · f3a06d2
1 parent 4352bf3
commit f3a06d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/scripts/train/anneal/OLMo2-32B-anneal.py b/src/scripts/train/anneal/OLMo2-32B-anneal.py
@@ -146,7 +146,7 @@ def build(
             launch=build_launch_config(
                 name=run_name,
                 root_dir=root_dir,
-                cmd=[script, cmd, run_name, cluster, *overrides],
+                cmd=[script, cmd, run_name, checkpoint, cluster, *overrides],
                 cluster=cluster,
                 nccl_debug=False,
             ),