Safetensors 文档
将权重转换为safetensors
您正在查看 main 版本,该版本需要从源代码安装。如果您想进行常规 pip 安装,请查看最新的稳定版本 (v0.5.0-rc.0)。
加入 Hugging Face 社区
并获得增强的文档体验
开始使用
将权重转换为 safetensors
PyTorch 模型权重通常使用 Python 的 pickle
实用程序保存并存储为 .bin
文件。为了以更安全的 safetensor
格式保存和存储您的模型权重,我们建议将您的权重转换为 .safetensors
。
转换模型权重最简单的方法是使用 Convert Space,前提是您的模型权重已存储在 Hub 上。Convert Space 下载 pickle 权重,转换它们,并打开一个 Pull Request 以将新转换的 .safetensors
文件上传到您的存储库。
对于较大的模型,Space 可能会有点慢,因为它的资源被绑定在转换其他模型上。您也可以尝试在本地运行 convert.py 脚本(这是 Space 正在运行的脚本)来转换您的权重。
如果 Space 有任何问题,请随时 ping @Narsil。