HuggingFace无法连接,用HF-mirror来绕开限制

在服务器上,通过sh文件运行vLLM服务时,报错连不上HuggingFace:

OSError: We couldn't connect to 'https://huggingface.co' to load this file, couldn't find it in the cached files and it looks like THUDM/chatglm2-6b is not the path to a directory containing a file named configuration_chatglm.py.

Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/transformers/installation#offline-mode'.

因为是实验室服务器,连不上HF是很正常的,我们用国内的菩萨网站 hf-mirror.com 就能解决这个问题:

先安装 huggingface_hub,这是 Hugging Face 官方提供的命令行工具,自带完善的下载功能:

pip install -U huggingface_hub

 然后在命令行设置环境变量,因为是临时性下载,我就没有把环境变量保存到.bashrc文件:

export HF_ENDPOINT=https://hf-mirror.com

 把HF_HUB_OFFLINE设置为0,表示从网上下载模型,如果为1,则表示从本地加载模型:

export HF_HUB_OFFLINE=0

我们打印一下看是不是正确的路径:

echo $HF_ENDPOINT
echo $HF_HUB_OFFLINE

 这个时候我们就可以从hf-mirror下载模型了,第一个THUDM/chatglm2-6b是hf-mirror的模型路径,第二个THUDM/chatglm2-6b是保存到的本地路径:

huggingface-cli download --resume-download THUDM/chatglm2-6b --local-dir THUDM/chatglm2-6b

此时就开始下载了:

等待完成后,我们可以到默认的.catch文件夹中查看下载的模型:

路径是:用户名/.cache/huggingface/hub

这样,我们就可以绕过防火墙的限制,下载HuggingFace上的模型了😆。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值