在使用 Hugging Face 下载模型时,由于网络原因可能会遇到下载速度慢或连接失败的问题。本文介绍几种配置国内镜像源的方法。
目录
方法一:安装 CLI 工具并设置环境变量
1. 安装 huggingface_hub
pip install -U huggingface_hub
2. 设置环境变量
export HF_ENDPOINT=https://hf-mirror.com
3. 下载模型示例
huggingface-cli download \
--local-dir-use-symlinks False \
microsoft/deberta-v3-large \
--local-dir /root/autodl-tmp/deberta_model \
--cache-dir /root/autodl-tmp/deberta_cache
方法二:修改源码配置
修改 constants.py 文件
找到你的 Python 环境中的 huggingface_hub
包位置:
通常在以下路径之一
/miniconda3/envs/环境名称/lib/python3.12/site-packages/huggingface_hub/constants.py
/usr/local/lib/python3.x/site-packages/huggingface_hub/constants.py
修改 constants.py
文件中的默认端点:
_HF_DEFAULT_ENDPOINT = "https://hf-mirror.com"
方法三:在代码中动态设置
在 Python 代码开头添加环境变量设置:
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
from huggingface_hub import hf_hub_download
可用的国内镜像源
- 官方镜像:
https://hf-mirror.com
- Alpha 镜像:
https://alpha.hf-mirror.com/
注意事项
- 设置环境变量后需要重启终端或重新加载环境
- 修改源码后需要重启 Python 解释器
- 建议优先使用方法一(环境变量),因为它不会影响其他项目
- 如果遇到权限问题,请确保有相应目录的写入权限
验证配置
配置完成后,可以通过以下命令验证是否生效:
huggingface-cli download --help
如果配置成功,下载速度应该会有明显提升。