解决Huggingface被墙下载模型的问题

参考:

如何快速下载huggingface模型------全方法总结 - padeoe的文章 - 知乎

https://zhuanlan.zhihu.com/p/663712983

和:

https://hf-mirror.com/docs/huggingface_hub/guides/download#download-from-the-cli

最近几月huggingface彻底无法访问,之前下载模型都是开梯子自己下载到本地再上传到服务器,这样实在比较麻烦,因此查资料之后得到解决方案:使用huggingface镜像站

Huggingface CLI

huggingface cli是hf官方的下载工具,使用pip安装:

cpp 复制代码
pip install -U huggingface_hub

安装好之后引入环境变量:

cpp 复制代码
export HF_ENDPOINT=https://hf-mirror.com

(只在当前终端生效,如果想长期生效改source文件)

然后就可以下载模型,这里下载的是facebook/bart-base,这里记住前面的facebook一定不要漏:

cpp 复制代码
huggingface-cli download --resume-download facebook/bart-base --local-dir /home/mguo/models/bart-base

后面的--local-dir就是指定下载路径(默认的在.~/cache/huggingface/hub 中)

为了统一管理,我就下载到统一的路径了,这样加载模型从本地加载:

cpp 复制代码
model_path = '/home/mguo/models/bart-base'
tokenizer = AutoTokenizer.from_pretrained(model_path, cache_dir=cache_dir)
相关推荐
Python大数据分析@15 分钟前
python操作CSV和excel,如何来做?
开发语言·python·excel
黑叶白树16 分钟前
简单的签到程序 python笔记
笔记·python
Shy96041829 分钟前
Bert完形填空
python·深度学习·bert
上海_彭彭40 分钟前
【提效工具开发】Python功能模块执行和 SQL 执行 需求整理
开发语言·python·sql·测试工具·element
zhongcx011 小时前
使用Python查找大文件的实用脚本
python
yyfhq2 小时前
sdnet
python
测试19982 小时前
2024软件测试面试热点问题
自动化测试·软件测试·python·测试工具·面试·职场和发展·压力测试
love_and_hope2 小时前
Pytorch学习--神经网络--搭建小实战(手撕CIFAR 10 model structure)和 Sequential 的使用
人工智能·pytorch·python·深度学习·学习
海阔天空_20132 小时前
Python pyautogui库:自动化操作的强大工具
运维·开发语言·python·青少年编程·自动化
零意@3 小时前
ubuntu切换不同版本的python
windows·python·ubuntu