一、.safetensors
是什么?
.safetensors
是一种由 Hugging Face 开发的新型文件格式,主要用于安全存储和高效加载深度学习模型中的张量(tensors)。
.safetensors
格式旨在解决传统文件格式(如 PyTorch 的 .pt
或 TensorFlow 的 .ckpt
)在安全性、加载速度和兼容性方面的不足。
.safetensors
是一种专为深度学习设计的新型文件格式,具有高效、安全、跨平台兼容等特点。
.safetensors
通过优化存储和加载机制,显著提升了模型文件的处理效率,并且由于其安全性设计,避免了传统格式中可能存在的安全隐患。
二、.safetensors
的特点:
- 安全性:
.safetensors
文件仅包含数值数据,不包含任何代码,从而避免了潜在的执行恶意代码的风险。这与使用 Python 的pickle
格式不同,后者可能被利用来执行任意代码。 - 高效性:该格式支持零拷贝加载,可以直接从磁盘加载到内存,无需额外解压或复制操作,显著提高了加载速度。
- 跨平台兼容性:
.safetensors</