开发验证

python3 - <<'PY'
from safetensors import safe_open 
from safetensors.numpy 
import save_file, load_file 
import safetensors 
import numpy as np

path = "/tmp/safetensors-test.safetensors"
arr = np.arange(12, dtype=np.float32).reshape(3, 4)
save_file({"weights": arr}, path, metadata={"format": "np"}) 
out = load_file(path)["weights"]  

with safe_open(path, framework="numpy") as f:     
    keys = list(f.keys())
    meta = f.metadata()
    loaded = f.get_tensor("weights")

print("safetensors_version=" + safetensors.__version__) 
print("numpy_version=" + np.__version__) 
print("keys=" + ",".join(keys)) 
print("metadata_format=" + meta.get("format", ""))
print("shape=" + str(out.shape))
print("sum=" + str(float(loaded.sum()))) 
assert safetensors.__version__ == "0.7.0"
assert keys == ["weights"] 
assert out.shape == (3, 4) 
assert float(loaded.sum()) == 66.0 
PY

safetensors_version=0.7.0
numpy_version=2.4.4
keys=weights
metadata_format=np
shape=(3, 4)
sum=66.0