您正在查看 主分支 版本,需要从源代码安装。如果您希望使用常规的 pip 安装,请查看最新的稳定版本(v0.3.2)。
NumPy API
safetensors.numpy.load_file
< 源代码 >( filename: Union ) → Dict[str, np.ndarray]
将 safetensors 文件加载到 NumPy 格式。
safetensors.numpy.load
< 源代码 >( data: bytes ) → Dict[str, np.ndarray]
从纯字节中将 safetensors 文件加载到 NumPy 格式。
safetensors.numpy.save_file
< 源代码 >( tensor_dict: Dict filename: Union metadata: Optional = None ) → None
将张量字典保存为 safetensors 格式的原始字节。
safetensors.numpy.save
< 源代码 >( tensor_dict: Dict metadata: Optional = None ) → bytes
将张量字典保存为 safetensors 格式的原始字节。