修改HuggingFace模型默认缓存路径

huggingface模型的默认缓存路径是~/.cache/huggingface/hub/

通常修改为自己的路径会更为方便。

方式一:cache_dir 参数

可以通过from_pretrained函数中的 cache_dir 参数来指定,缺点,每次都需要手动指定,比较麻烦。

如:加载一个bert-base-uncased模型

python 复制代码
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 环境变量

bash 复制代码
export HF_HOME=~/leo038/cache/
相关推荐
码界奇点4 分钟前
豆包新模型矩阵与PromptPilot构建企业级AI开发的体系化解决方案
人工智能·线性代数·ai·语言模型·矩阵·硬件工程
YangYang9YangYan9 分钟前
2025年跨领域管理能力提升认证路径分析
大数据·人工智能
那雨倾城23 分钟前
PiscCode轨迹跟踪Mediapipe + OpenCV进阶:速度估算
图像处理·人工智能·python·opencv·计算机视觉
兮希yxx39 分钟前
conda配置pytorch虚拟环境
人工智能·pytorch·conda
zhongtianhulian1 小时前
中天互联:AI 重塑制造,解锁智能生产新效能
人工智能·制造
renhongxia11 小时前
大模型赋能电子制造全生命周期质量管理的应用及实践
人工智能
Edward.W1 小时前
如何制造一个AI Agent:从“人工智障”到“人工智能”的奇幻漂流
人工智能·制造
NingboWill1 小时前
AI日报 - 2025年09月05日
人工智能·microsoft
SheldonChang1 小时前
Onlyoffice集成与AI交互操作指引(Iframe版)
java·人工智能·ai·vue·onlyoffice·postmessage
云卓SKYDROID1 小时前
无人机报警器8G信号技术解析
人工智能·无人机·航电系统·高科技·云卓科技