在 ModelScope 中,snapshot_download 是一个非常有用的工具,可以直接下载模型文件到指定路径,而无需加载模型。以下是使用 snapshot_download 下载模型的具体方法:
1. 安装 ModelScope SDK
pip install modelscope
2. 使用 snapshot_download 下载模型
from modelscope import snapshot_download
# 指定模型名称
model_name = "Qwen/Qwen1.5-1.8B-Chat"
# 指定下载路径(绝对路径)
custom_path = "/home/llm/ai/model"
# 下载模型
model_dir = snapshot_download(
model_name, # 模型名称
cache_dir=custom_path, # 指定下载路径
revision="master" # 可选:指定模型版本(默认master)
)
print(f"模型已下载到:{model_dir}")