Safetensors 文档

将权重转换为 safetensors

您正在查看的是需要从源码安装。如果您想进行常规的 pip 安装,请查看最新的稳定版本 (v0.5.0-rc.0)。
Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

将权重转换为 safetensors

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

如果您的模型权重已经存储在 Hub 上,最简单的转换方法是使用 转换 Space。转换 Space 会下载 pickled 格式的权重,进行转换,并开启一个拉取请求(Pull Request)以上传新转换的 .safetensors 文件到您的仓库。

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

如果遇到任何有关 Space 的问题,请随时联系 @Narsil

< > 在 GitHub 上更新