修改hugging face cache目录

hugging face预训练模型都特别大,系统tmp空间很可能不够

这时候需要修改cache到自定义的还有空间的目录。

官网上这么解释的,我不再作过多解释:
在这里插入图片描述
把以下的代码在终端运行,或者加入bashrc中

# /my_dir/tmp是我设置的新的缓存目录,你需要修改成你自己的目录
export HF_HOME=/my_dir/tmp
export XDG_CACHE_HOME=/my_dir/tmp
export TRANSFORMERS_CACHE=/my_dir/tmp

参考文献

[1] huggingface官网

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐