fix ckpt file

pytorch · Dec 3, 2024 · 623a955 · 623a955
1 parent e544031
commit 623a955
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes/configs/llama3_2/1B_lora_single_device.yaml b/recipes/configs/llama3_2/1B_lora_single_device.yaml
@@ -36,7 +36,7 @@ checkpointer:
   _component_: torchtune.training.FullModelHFCheckpointer
   checkpoint_dir: /tmp/Llama-3.2-1B-Instruct/
   checkpoint_files: [
-     epoch_2/ft-model-00001-of-00001.safetensors
+     model.safetensors
   ]
   recipe_checkpoint: null
   output_dir: ${output_dir}