Skip to content

Commit e49fe94

Browse files
committed
Comment
1 parent ce6e987 commit e49fe94

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/qwen2_5/convert_weights.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def qwen_2_tune_to_meta(state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.
5252
return converted_state_dict
5353

5454

55-
# TODO: no need to use TorchTune checkpointer, can just aggregate checkpoint files by ourselves.
55+
# Don't necessarily need to use TorchTune checkpointer, can just aggregate checkpoint files by ourselves.
5656
checkpointer = FullModelHFCheckpointer(
5757
checkpoint_dir="/home/jackzhxng/.cache/huggingface/hub/models--Qwen--Qwen2.5-1.5B/snapshots/8faed761d45a263340a0528343f099c05c9a4323/",
5858
checkpoint_files=["model.safetensors"],

0 commit comments

Comments
 (0)