examples/tutorials/grpo/step_1_download_model_ms.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
或使用命令行

python3 step_1_download_model_ms.py \
--repo_id Qwen/Qwen2.5-3B-Instruct \
--local_dir /root/autodl-tmp/OpenMiniMind/pretrained_models/Qwen/Qwen2.5-3B-Instruct

"""
import argparse
import os
from pathlib import Path
import platform

if platform.system() in ("Windows", "Darwin"):
    from project_settings import project_path, temp_directory
else:
    project_path = os.path.abspath("../../../")
    project_path = Path(project_path)
    temp_directory = Path("/root/autodl-tmp/OpenMiniMind/temp")

from modelscope import snapshot_download


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--repo_id", default="Qwen/Qwen2.5-3B-Instruct", type=str)
    parser.add_argument(
        "--local_dir",
        default=(temp_directory / "../pretrained_models/Qwen/Qwen2.5-3B-Instruct").as_posix(),
        type=str
    )
    args = parser.parse_args()
    return args


def main():
    args = get_args()

    snapshot_download(
        model_id=args.repo_id,
        local_dir=args.local_dir,
    )
    return


if __name__ == "__main__":
    main()