之前分享过通过镜像网站手工下载模型文件的技巧(见这里模型的手工下载技巧-镜像网站的使用)。但有的时候,模型文件数量较多,一个个手工下载非常不便。比如著名的“麦橘写实”模型。
有没有什么好办法可以把整个目录都下载下来呢?
# 以下代码为程序运行进行设置
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
from huggingface_hub import snapshot_download
# 替换为自己的存储目录名称
model_path = "majicMIX_realistic_v7"
snapshot_download(
repo_id="digiplay/majicMIX_realistic_v7",
local_dir=model_path,
max_workers=4, # 指定同时下载的线程数
resume_download=True # 支持断点续传
)
直接运行该程序,耐心等待。无论多大的模型库都能下载下来。