safetensors

Simple, safe way to store and distribute tensors

APACHE-2.0 License

Downloads
20M
Stars
2.4K
Committers
47

Bot releases are visible (Hide)

safetensors - Python v0.2.7

Published by Narsil almost 2 years ago

safetensors - Rust 0.2.6

Published by Narsil almost 2 years ago

First Rust release.

safetensors - Python v0.2.6

Published by Narsil almost 2 years ago

  • Python 3.11 Linux support.
safetensors - Python v0.2.5

Published by Narsil almost 2 years ago

  • [#102] Hotfix for Windows i32 overflow within PyO3
  • Some quality of life (better error messages, improved convert.py to be more general.
  • Python 3.11 prebuilt binary
safetensors - Python 0.2.4

Published by Narsil almost 2 years ago

Fast CPU load on Pytorch
Fast GPU load on Python (unsafe, need to use SAFETENSORS_FAST_GPU=1 environement variable to use.

safetensors - Python v0.2.3

Published by Narsil almost 2 years ago

Remove importib. This was bogus and it crashes on Python2.7.

safetensors - Python v0.2.2

Published by Narsil almost 2 years ago

  • Adding device argument to safe_open to reduce overhead of sending tensors to GPU.
safetensors - Python v0.2.1

Published by Narsil about 2 years ago

math.prod is not supported for Python < 3.8. Putting back np.prod in there.

safetensors - Python-v0.2.0

Published by Narsil about 2 years ago

Added safe_open.
Added __metadata__ special key in the format (ability to store layout information that could be useful to transformers.)
Removed some API surface.
Cleaned up TF to be actually TF and not jax anymore.

Very shallow changelog, this is still a lot in exploration mode.

safetensors - Python v0.0.1

Published by Narsil about 2 years ago

Initial release with barebone features just to get a PoC running.

Main features missing:

  • Lazy loading
  • DOS protection.