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

pystse-safetensors

Provides functions to read and write safetensors which aim to be safer than their PyTorch counterpart. The format is 8 bytes which is an unsized int, being the size of a JSON header, the JSON header refers the dtype the shape and data_offsets which are the offsets for the values in the rest of the file

1 unstable release

new 0.0.1 Apr 23, 2024

#10 in #huggingface


Used in pystse

Apache-2.0

65KB
1.5K SLoC

pystse-safetensors

参见 Pystse

Dependencies

~0.7–1.4MB
~33K SLoC