Skip to content

Commit 739d42b

Browse files
ebsmotherspbontrager
authored andcommitted
update teacher checkpointer paths in KD config (#2496)
1 parent 0ffc681 commit 739d42b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

recipes/configs/qwen2/1.5_to_0.5B_KD_lora_distributed.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,9 @@ checkpointer:
4646

4747
teacher_checkpointer:
4848
_component_: torchtune.training.FullModelHFCheckpointer
49-
checkpoint_dir: /tmp/Qwen2-1.5B-Instruct-lora-finetune
49+
checkpoint_dir: /tmp/Qwen2-1.5B-Instruct
5050
checkpoint_files: [
51-
hf_model_0001_0.pt
51+
model.safetensors
5252
]
5353
recipe_checkpoint: null
5454
output_dir: ${output_dir}

0 commit comments

Comments
 (0)