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 ca1d6c0 commit 8b51959Copy full SHA for 8b51959
examples/models/qwen2_5/convert_weights.py
@@ -68,7 +68,6 @@ def main():
68
69
# Don't necessarily need to use TorchTune checkpointer, can just aggregate checkpoint files by ourselves.
70
checkpointer = FullModelHFCheckpointer(
71
- # checkpoint_dir="/home/jackzhxng/.cache/huggingface/hub/models--Qwen--Qwen2.5-1.5B/snapshots/8faed761d45a263340a0528343f099c05c9a4323/",
72
checkpoint_dir=args.input_dir,
73
checkpoint_files=["model.safetensors"],
74
output_dir=".",
@@ -80,7 +79,6 @@ def main():
80
79
81
print("Converting checkpoint...")
82
sd = qwen_2_tune_to_meta(sd["model"])
83
- # torch.save(sd, "/home/jackzhxng/models/qwen2_5-1_5b.pth")
84
85
torch.save(sd, args.output)
86
print(f"Checkpoint saved to {args.output}")
0 commit comments