修改HuggingFace模子默认缓存路径

huggingface模子的默认缓存路径是~/.cache/huggingface/hub/
通常修改为自己的路径会更为方便。
方式一：cache_dir 参数

可以通过from_pretrained函数中的 cache_dir 参数来指定，缺点，每次都必要手动指定，比力麻烦。
如：加载一个bert-base-uncased模子

1. from transformers import AutoTokenizer, AutoModelForMaskedLM
2. tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased",cache_dir="路径")
3. model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-uncased",cache_dir="路径")

复制代码

方式二： 设置HF_HOME 环境变量

1. export HF_HOME=~/leo038/cache/

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。