examples/tutorials/by_deepspeed/step_1_download_hub.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
或使用命令行
pip install modelscope
modelscope download \
--model 'qgyd2021/Qwen3-8B-sft-deepspeed' \
--local_dir '/root/autodl-tmp/trained_models/Qwen3-8B-sft-deepspeed'

"""
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="qgyd2021/Qwen3-8B-sft-deepspeed", type=str)
    parser.add_argument(
        "--local_dir",
        default=(temp_directory / "trained_models/Qwen3-8B-sft-deepspeed").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()