大模型下载到本地

一、huggingface

• 方式一

from huggingface_hub import snapshot_download

local_dir = "./origin"
model_name = "Qwen/Qwen2.5-1.5B"

# snapshot_download(repo_id=model_name, cache_dir=local_dir)
model_dir = snapshot_download(
    model_name,
    cache_dir=local_dir
)
print(f"模型路径: {model_dir}")

二、modelscope

• 方式一

# model_download.py
import os
import torch
from modelscope import snapshot_download, AutoModel, AutoTokenizer
model_dir = snapshot_download('qwen/Qwen2-7B-Instruct', cache_dir='/root/autodl-tmp', revision='master')
print(f"模型路径: {model_dir}")