Safetensors

您正在查看主分支版本，它需要从源代码安装。如果您想要常规的 pip 安装，请查看最新的稳定版本（v0.3.2）。

加入 Hugging Face 社区

并获得增强型文档体验

协作模型、数据集和空间

使用加速推理获得更快的示例

在文档主题之间切换

开始使用

PaddlePaddle API

safetensors.paddle.load_file

< 源代码 >

( filename: Union device = 'cpu' ) → Dict[str, paddle.Tensor]

参数

filename (str, 或 os.PathLike)) — 包含张量的文件名称
device (Union[Dict[str, any], str], 可选，默认为 cpu) — 加载后张量需要位于的设备。可用选项是所有常规的 paddle 设备位置

返回值

Dict[str, paddle.Tensor]

包含名称作为键，值作为 paddle.Tensor 的字典

将 safetensors 文件加载到 paddle 格式。

示例

from safetensors.paddle import load_file

file_path = "./my_folder/bert.safetensors"
loaded = load_file(file_path)

safetensors.paddle.load

< 源代码 >

( data: bytes device: str = 'cpu' ) → Dict[str, paddle.Tensor]

参数

data (bytes) — safetensors 文件的内容

返回值

Dict[str, paddle.Tensor]

包含名称作为键，值作为 paddle.Tensor 在 cpu 上的字典

从纯字节中将 safetensors 文件加载到 paddle 格式。

示例

from safetensors.paddle import load

file_path = "./my_folder/bert.safetensors"
with open(file_path, "rb") as f:
    data = f.read()

loaded = load(data)

safetensors.paddle.save_file

< 源代码 >

( tensors: Dict filename: Union metadata: Optional = None ) →

None

将张量字典保存为 safetensors 格式的原始字节。

示例

from safetensors.paddle import save_file
import paddle

tensors = {"embedding": paddle.zeros((512, 1024)), "attention": paddle.zeros((256, 256))}
save_file(tensors, "model.safetensors")

safetensors.paddle.save

< 源代码 >

( tensors: Dict metadata: Optional = None ) → bytes

参数

tensors (Dict[str, paddle.Tensor]) — 输入张量。张量需要是连续且密集的。
metadata (Dict[str, str], 可选, 默认为 None) — 您可能希望保存在标题中的可选纯文本元数据。例如，它可以用于指定有关底层张量的更多信息。这纯粹是信息性的，不会影响张量的加载。

返回值

字节

表示该格式的原始字节

将张量字典保存为 safetensors 格式的原始字节。

示例

from safetensors.paddle import save
import paddle

tensors = {"embedding": paddle.zeros((512, 1024)), "attention": paddle.zeros((256, 256))}
byte_data = save(tensors)

< > 在 GitHub 上更新

←Tensorflow API Flax API→