dimitri/bqn-safetensors
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add test scripts

Browse source
This commit is contained in:
Dimitri Lozeve 2024-11-15 20:59:02 +01:00
parent 6c1d57f9cd
commit 8943455447
4 changed files with 60 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
loadtest.py Normal file
View file

@ -0,0 +1,10 @@
# /// 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)