安装与验证

python3 -m pip install --upgrade pip setuptools wheel 
python3 -m pip install safetensors==0.7.0

python3 -m pip install numpy==2.4.4

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