IT评测·应用市场-qidao123.com

标题: 修改HuggingFace模子默认缓存路径 [打印本页]

作者: 缠丝猫 时间: 2025-3-20 03:20
标题: 修改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企服之家，中国第一个企服评测及商务社交产业平台。

欢迎光临 IT评测·应用市场-qidao123.com (https://dis.qidao123.com/)