bqn-safetensors/loadtest.py
2024-11-15 20:59:02 +01:00

10 lines
213 B
Python

# /// script
# dependencies = ["numpy", "safetensors"]
# ///
from safetensors.numpy import load_file
arrs = load_file("test.safetensors")
for name, arr in arrs.items():
print(name, arr.dtype)
print(arr)