#tensor #huggingface #tensorflow #pytorch #safetensors #read-write

pystse-safetensors

提供读取和写入 safetensors 的函数,旨在比 PyTorch 对应版本更安全。格式为 8 字节的无尺寸 int,表示 JSON 头的大小,JSON 头引用了 dtypeshapedata_offsets,这些是文件其余部分中值的偏移量。

1 个不稳定版本

0.0.1 2024 年 4 月 23 日

#12 in #huggingface


pystse 中使用

Apache-2.0

65KB
1.5K SLoC

pystse-safetensors

参见 Pystse

依赖项

~0.7–1.6MB
~36K SLoC