计算huggingface模型占用硬盘空间的实战代码

  大家好,我是herosunly。985院校硕士毕业,现担任算法研究员一职,热衷于机器学习算法研究与应用。曾获得阿里云天池比赛第一名,CCF比赛第二名,科大讯飞比赛第三名。拥有多项发明专利。对机器学习和深度学习拥有自己独到的见解。曾经辅导过若干个非计算机专业的学生进入到算法行业就业。希望和大家一起成长进步。

  本文主要介绍了计算huggingface模型占用硬盘空间的实战代码,希望对学习大语言模型的同学们有所帮助。

文章目录

  • 1. 前言
  • 2. 代码思路
  • 3. 实战代码

1. 前言

  最近在和学习大模型的同学们交流,由于大模型占用的磁盘空间都比较大,所以经常会出现磁盘空间接近爆满的情况。

计算huggingface模型占用硬盘空间的实战代码_第1张图片

  因此如果能在下载模型之前预留好对应的磁盘空间,就能够很好的解决上述的问题。

  在经过了亲身的实践后,终于找到了可复现的实战代码,最终将详细的代码总结如下。希望对同学们有所帮助。

你可能感兴趣的:(类GPT工具使用技巧和论文解读,huggingface,模型空间,计算模型空间,实战代码)