当前位置: 首页 > article >正文

修改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/

http://www.kler.cn/a/596265.html

相关文章:

  • 【yolo】使用 Netron 可视化深度学习模型:从 YOLOv1 到 YOLOv8 的探索
  • 【css酷炫效果】纯CSS实现球形阴影效果
  • 基于springbo校园安全管理系统(源码+lw+部署文档+讲解),源码可白嫖!
  • IoTDB日志提示Too many open files
  • mac环境下chatwoot客服聊天docker本地部署+对接通义千问Qwen2.5
  • C++代码2-多目标算法求解车辆路径规划
  • 使用PlotNeuralNet绘制ResNet50模型
  • Prometheus Exporter系列-Mysql_Exporter一键部署
  • 面试中JVM常被问到的问题以及对应的答案
  • 优化广告投放算法
  • php的用途和基础语法【初学者进】
  • unity npc射线检测碰撞
  • 《前端秘籍:JS语法+DOM操作,从新手到高手》
  • 代码随想录算法训练营第十五天 | 数组 |长度最小的子数组和螺旋矩阵II
  • 【ESP32】虚拟机Ubuntu20.04下ESP32环境搭建
  • 基于deepseek的智能语音客服【第二讲】后端异步接口调用封装
  • 在 Elasticsearch 中扩展后期交互模型 - 第 2 部分 - 8.18
  • CTF题目《easy_tornado》(护网杯 2018)Write Up
  • Flink相关面试题
  • 分析K8S中Node状态为`NotReady`问题