test checkpoints

stanford-crfm · Apr 24, 2024 · 7ed76fc · 7ed76fc
1 parent a5e4b4c
commit 7ed76fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/config/for_llama2_7b_git.yaml b/config/for_llama2_7b_git.yaml
@@ -8,8 +8,8 @@ data:
 model:
   type: llama
 # TODO: uncomment this once we resolve the resource exhaustion issue
-hf_checkpoint: "levanter-checkpoints/llama2-trace-1b-seed42/19zqlxdg/step-200/model"
-second_hf_checkpoint: "levanter-checkpoints/llama2-trace-1b-seed42/19zqlxdg/step-200/model"
+hf_checkpoint: "gs://levanter-checkpoints/llama2-trace-1b-seed42/19zqlxdg/step-200/model"
+second_hf_checkpoint: "gs://levanter-checkpoints/llama2-trace-1b-seed42/19zqlxdg/step-200/model"
 
 trainer:
   wandb: