如何使用huggingface下载数据集和预训练模型

        如果各位在下载huggingface上的模型和数据库也会出现“connect closed/failed”等错误,不妨试试下面的解决方案,思路大致是,通过设置镜像的方式来解决。

下载数据集

1. 找到你要下载的数据库名称,并复制

2. 打开终端,并选择需要使用的conda环境,编写bash文件(或者自己逐行也行)

#安装
pip install -U huggingface_hub
#设置镜像
export HF_ENDPOINT=https://hf-mirror.com
#数据库下载设置
huggingface-cli download --repo-type dataset --resume-download liuhaotian/LLaVA-CC3M-Pretrain-595K --local-dir dataset  --local-dir-use-symlinks False

一共上述3行,安装第三方库 ----> 设置环境 ----> 下载数据库

其中:

--repo-type dataset:下载类型为数据库

--resume-download liuhaotian/LLaVA-CC3M-Pretrain-595K  这个是需要下载的数据库名称

--local-dir dataset 保存的文件夹名称

--local-dir-use-symlinks False 关闭软连接,网站上数据库什么样下载下来就是什么样

3. 终端出现下载条说明开始下载了。

下载预训练模型

同理,和下载数据库主要的区别就是下载命令参数有些不同,这里直接上两个下载链接,一个是CLIP 一个是 Qwen-4B的示例

pip install -U huggingface_hub
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --resume-download openai/clip-vit-large-patch14-336 --local-dir ./openai/clip-vit-large-patch14-336 --local-dir-use-symlinks False
huggingface-cli download --resume-download Qwen/Qwen1.5-4B --local-dir ./Qwen1.5-4b-Chat --local-dir-use-symlinks False

  • 9
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值