reasoning-gym/training/qwen-math/scripts/set/run_download_model.py
Zafir Stojanovski 0cda6b1205
qwen math training code (#435)
* qwen math training code

* pre-commit
2025-05-16 13:19:19 +02:00

9 lines
274 B
Python

import os
from huggingface_hub import snapshot_download
if __name__ == "__main__":
CKPT_DIR = os.environ["CKPT_DIR"]
print("Downloading model ...")
snapshot_download(repo_id="Qwen/Qwen2.5-3B-Instruct", local_dir=f"{CKPT_DIR}/models/Qwen2.5-3B-Instruct/base")