您正在查看 主 版本,它需要从源代码安装. 如果你想要常规的 pip 安装,请查看最新的稳定版本 (v0.3.2).
Tensorflow API
safetensors.tensorflow.load_file
< 源代码 >( filename: Union ) → Dict[str, tf.Tensor]
将 safetensors 文件加载到 tensorflow 格式中。
safetensors.tensorflow.load
< 源代码 >( data: bytes ) → Dict[str, tf.Tensor]
从纯字节中将 safetensors 文件加载到 tensorflow 格式中。
safetensors.tensorflow.save_file
< 源代码 >( tensors: Dict filename: Union metadata: Optional = None ) → None
将张量字典保存到 safetensors 格式的原始字节中。
safetensors.tensorflow.save
< 源代码 >( tensors: Dict metadata: Optional = None ) → bytes
将张量字典保存到 safetensors 格式的原始字节中。