We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce6e987 commit e49fe94Copy full SHA for e49fe94
examples/models/qwen2_5/convert_weights.py
@@ -52,7 +52,7 @@ def qwen_2_tune_to_meta(state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.
52
return converted_state_dict
53
54
55
-# TODO: no need to use TorchTune checkpointer, can just aggregate checkpoint files by ourselves.
+# Don't necessarily need to use TorchTune checkpointer, can just aggregate checkpoint files by ourselves.
56
checkpointer = FullModelHFCheckpointer(
57
checkpoint_dir="/home/jackzhxng/.cache/huggingface/hub/models--Qwen--Qwen2.5-1.5B/snapshots/8faed761d45a263340a0528343f099c05c9a4323/",
58
checkpoint_files=["model.safetensors"],
0 commit comments