Safetensors 文档

将权重转换为safetensors

您正在查看 main 版本,该版本需要从源代码安装。如果您想进行常规 pip 安装,请查看最新的稳定版本 (v0.5.0-rc.0)。
Hugging Face's logo
加入 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

< > 在 GitHub 上更新