Safetensors 文档

将权重转换为 safetensors

您正在查看 版本,该版本需要从源代码安装. 如果你想要常规的 pip 安装,请查看最新的稳定版本 (v0.3.2).
Hugging Face's logo
加入 Hugging Face 社区

并获得增强文档体验

以开始

将权重转换为 safetensors

PyTorch 模型权重通常以 .bin 文件的形式保存和存储,并使用 Python 的 pickle 实用程序。要以更安全的 safetensor 格式保存和存储您的模型权重,我们建议将您的权重转换为 .safetensors

将模型权重转换为 .safetensors 的最简单方法是使用 Convert Space,前提是您的模型权重已存储在 Hub 上。Convert Space 会下载腌制的权重,将其转换,并打开一个 Pull Request 以将新转换的 .safetensors 文件上传到您的存储库。

对于较大的模型,Space 可能有点慢,因为它的资源被用于转换其他模型。您也可以尝试在本地运行 convert.py 脚本(这就是 Space 正在运行的内容)来转换您的权重。

如果您在使用 Space 时遇到任何问题,请随时联系 @Narsil

< > 更新 在 GitHub 上