使用镜像站点修改 Hugging Face Hub 默认网址
1. 使用 pip show
查找 Hugging Face Hub 包的位置
你可以使用 pip show
命令来找到 huggingface_hub
包的安装位置:
pip show huggingface_hub
输出结果会包含类似以下的信息:
Name: huggingface-hub
Version: 0.x.x
Location: /usr/local/lib/python3.x/dist-packages
在 Location
字段中显示的路径即为 huggingface_hub
包的安装位置。在该路径下,可以找到 constants.py
文件。
2. 修改 constants.py
文件
找到 huggingface_hub
包的安装路径后,进入到对应的目录并打开 constants.py
文件。你需要修改默认的 Hugging Face 网址 _HF_DEFAULT_ENDPOINT
,将其更改为你希望使用的镜像网址。例如,假设你有一个镜像站点 https://mirror-hf.co
,可以按以下方式修改:
# 原代码
# _HF_DEFAULT_ENDPOINT = "https://huggingface.co"
# 修改为镜像网址
_HF_DEFAULT_ENDPOINT = "https://mirror-hf.co"
保存修改并退出。
3. 重启 Python 环境
修改完成后,确保 Python 环境重新加载了这些更改。可以通过以下方式重启 Python 环境:
- 重启终端或 Jupyter Notebook
- 重新运行 Python 脚本
python3 script_name.py
4. 测试
在完成上述修改之后,重新运行你的代码,测试 Hugging Face 是否能通过镜像站点正确加载模型。此时应该能够避免频繁的超时错误。
通过这些步骤,可以方便地使用镜像站点加速访问 Hugging Face,特别是在网络条件不理想的情况下。