使用镜像源下载Hugging Face模型

        使用国内的网络甚至使用科学上网在hugging face中transformer上下载模型参数往往会失败,这个时候就需要我们手动去hugging face的model hub中下载需要的模型参数,这样十分费时费力。本博客旨在解决这个问题,使研究者们在使用模型下载模型参数上少花时间,将更多的精力花费在更有价值的地方。

方法:

主要参考

(11 封私信 / 78 条消息) 如何看待huggingface.co已无法访问? - 知乎 (zhihu.com)

我结合博文以及评论大神给的代码,构建了属于自己的下载模型方法。

1. 下载git lfs

这个东西用来下载大的模型权重文件,比如几个G的pytorch_model.bin文件,我的这个场景是linux,不知道windows需不需要。下载这个东西的方法有很多,搜索git lfs安装一大堆,这里我是在实验室服务器上,没有sudo权限,用的和这个从local user安装到使用的git-lfs采坑小结 - 一个兔子洞 | Y's Blog (hua-ys.github.io)差不多的方法。主要的点是1.将install.sh文件prefix后面的路径修改为local user下的自定义路径 2.bash xxx/install.sh之后一定要注意要有export PATH=$PATH:your_path这个命令,比如你install xxx/install.sh之后,安装到了自定义的路径/home/xxx/git-lfs/bin/git-lfs,要在终端使用export PATH=$PATH:/home/xxx/git-lfs/bin导入到全局变量里(这是我踩过的坑和大家说一下)

2. 在要下载模型配置的文件夹下使用 git lfs install

这一步的命令是为了把这个文件夹变成一个git文件夹,便于从网上拉取文件

3. 使用脚本下载模型文件

脚本在这里Command-line Tool for Easy Downloading of Huggingface Models (github.com)

我直接跑这个脚本因为网络的原因跑不通,需要对脚本稍作修改,改成下图所示,使用镜像网站。

图1 修改hdf.sh文件

 然后使用下面这条命令,在终端里跑

xxx/hdf.sh meta-llama/Llama-2-7b --hf_username xxx --hf_token xxx

这里给了个例子,把上面的“meta-llama/Llama-2-7b”换成你要的模型配置路径, hf_username换成你hugging face账号的user name,hf_token换成你账号中的token,在这里可以查看到Hugging Face – The AI community building the future.

通过上面的步骤就可以下载模型啦!希望这篇博客在帮助我自己回忆解决方案的同时,对大家也有点帮助!

  • 4
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值