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