huggingface已经是当前广泛使用的模型平台,很多开源模型都存放在huggingface。这里记录一下比较简单使用的python下载方法,如果想了解得更全面,可以参考这篇文章:如何快速下载huggingface模型——全方法总结
这里使用snapshot_download
方式,非常简单直接:
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
from huggingface_hub import snapshot_download
# 下载整个模型的所有文件
local_dir = snapshot_download(repo_id="Qwen/Qwen2.5-VL-7B-Instruct", local_dir="/ssd/wuyuming/LLM-modelzoo/Qwen/Qwen2.5-VL-7B-Instruct")
print(f"Model directory downloaded to: {local_dir}")
其中,repo_id
是Hugging Face
上模型id,如Qwen/Qwen2.5-VL-7B-Instruct
;local_dir
指定模型保存的目录。