计算huggingface模型占用硬盘空间的实战代码
大家好，我是herosunly。985院校硕士毕业，现担任算法研究员一职，热衷于机器学习算法研究与应用。曾获得阿里云天池比赛第一名，CCF比赛第二名，科大讯飞比赛第三名。拥有多项发明专利。对机器学习和深度学习拥有自己独到的见解。曾经辅导过若干个非计算机专业的学生进入到算法行业就业。希望和大家一起成长进步。
本文主要介绍了计算huggingface模型占用硬盘空间的实战代码，希望对学习大语言模型的同学们有所帮助。
文章目录
- 1. 前言
- 2. 代码思路
- 3. 实战代码
1. 前言
最近在和学习大模型的同学们交流，由于大模型占用的磁盘空间都比较大，所以经常会出现磁盘空间接近爆满的情况。
因此如果能在下载模型之前预留好对应的磁盘空间，就能够很好的解决上述的问题。
在经过了亲身的实践后，终于找到了可复现的实战代码，最终将详细的代码总结如下。希望对同学们有所帮助。