还在用torch.save()？快来看看safetensors吧！_safetensors.torch

1. 前言

Safetensors 是一个由 HuggingFace 开发的库（采用 Rust 编写），用于安全地存储和加载张量（tensors）。与其他序列化格式（如 pickle）相比，Safetensors 提供了更快的加载速度和更高的安全性。这个库特别适用于深度学习模型的序列化，尤其是在需要高效加载大型模型权重时。

安装：

pip install safetensors
1

2. safetensors.torch

这一节主要讲解safetensors中有关PyTorch的API。

2.1 读取

2.1.1 load_file

函数签名如下：

def load_file(filename: Union[str, os