|
@@ -46,7 +46,6 @@ model:
|
|
|
checkpointer:
|
|
|
_component_: torchtune.training.FullModelHFCheckpointer
|
|
|
checkpoint_dir: /tmp/Meta-Llama-Guard-3-8B/
|
|
|
- #checkpoint_dir: /home/agunapal/fork/torchtune/ckpt/llama_guard/
|
|
|
checkpoint_files: [
|
|
|
model-00001-of-00004.safetensors,
|
|
|
model-00002-of-00004.safetensors,
|
|
@@ -54,7 +53,6 @@ checkpointer:
|
|
|
model-00004-of-00004.safetensors
|
|
|
]
|
|
|
recipe_checkpoint: null
|
|
|
- #recipe_checkpoint: /tmp/torchtune/llama_guard_3_8B/full/recipe_state/recipe_state.pt
|
|
|
output_dir: ${output_dir}
|
|
|
model_type: LLAMA3
|
|
|
resume_from_checkpoint: False
|