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