safetensors是什么文件

Safetensors 是一种用于安全存储和加载张量数据的文件格式。

简介

• 安全性：Safetensors 旨在替代 Python 的 pickle 模块，后者不安全，可能包含恶意代码。2
• 快速性：它支持零拷贝操作，保证了加载速度。4
• 元数据信息：Safetensors 文件包含模型权重的元数据信息，如数据类型、形状和数据偏移量。2

特点

• 新型格式：由 Hugging Face 推出，特别关注模型安全性、隐私保护。1
• 不包含执行代码：与 pickle 相比，Safetensors 不包含执行代码，提高了加载时的安全性。6
• 适用于 PyTorch：Safetensors 可用于 PyTorch 框架，安全地保存和加载模型权重。3

应用

• 存储模型权重：常用于存储大型深度学习模型的权重参数。1
• 与 Hugging Face 生态系统集成：特别适合在 Hugging Face 生态中快速部署模型。1

误区澄清

• 与 TensorFlow Lite 的关系：Safetensors 不是由谷歌开发，也不是 TensorFlow Lite 的模型文件格式。8