「 AI 」 二月 16, 2025
可本地部署使用的大模型显存资源估算工具
文章字数 3.5k 阅读约需 3 mins.
🤗 Model Memory Calculator 是 🤗 Accelerate 库提供的一个模型显存计算工具,可估算模型训练或推理时所需的显存大小。
但目前该在线工具无法正常使用,即使使用默认的模型名称,也会报在 Hub 中无法找到该模型:
在该 space 的 discussions 中也有不少人遇到了此问题。
本文提供一种本地化部署运行此工具的方法,还可通过指定 HF_ENDPOINT
避免无法访问 Hugging Face 的问题。
直接部署 https://huggingface.co...
查看全文