修改HuggingFace模子默认缓存路径
huggingface模子的默认缓存路径是~/.cache/huggingface/hub/通常修改为自己的路径会更为方便。
方式一：cache_dir 参数
可以通过from_pretrained函数中的 cache_dir 参数来指定，缺点，每次都必要手动指定，比力麻烦。
如：加载一个bert-base-uncased模子
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-uncased",cache_dir="路径")
model = AutoModelForMaskedLM.from_pretrained("google-bert/bert-base-uncased",cache_dir="路径")
方式二： 设置HF_HOME 环境变量
export HF_HOME=~/leo038/cache/
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。
页:
[1]