Safetensors
您正在查看的是需要从源码安装。如果您想通过 pip 进行常规安装,请查看最新的稳定版本 (
v0.5.0-rc.0)。
加入 Hugging Face 社区
并获得增强的文档体验
Flax API
safetensors.flax.load_file
< 源代码 > ( filename: typing.Union[str, os.PathLike] ) → Dict[str, Array]
参数
- filename (
str
, 或 os.PathLike
) — 包含张量的文件名
一个字典,其中键为名称,值为 Array
将 safetensors 文件加载为 flax 格式。
示例
from safetensors.flax import load_file
file_path = "./my_folder/bert.safetensors"
loaded = load_file(file_path)
safetensors.flax.load
< 源代码 > ( data: bytes ) → Dict[str, Array]
参数
- data (
bytes
) — safetensors 文件的内容
一个字典,其中键为名称,值为 CPU 上的 Array
从纯字节数据加载 safetensors 文件为 flax 格式。
示例
from safetensors.flax import load
file_path = "./my_folder/bert.safetensors"
with open(file_path, "rb") as f:
data = f.read()
loaded = load(data)
safetensors.flax.save_file
< 源代码 > ( tensors: typing.Dict[str, jax.Array] filename: typing.Union[str, os.PathLike] metadata: typing.Optional[typing.Dict[str, str]] = None ) → None
参数
- tensors (
Dict[str, Array]
) — 待保存的张量。张量需要是连续且密集的。 - filename (
str
, or os.PathLike
)) — 我们要保存到的文件名。 - metadata (
Dict[str, str]
, 可选, 默认为 None
) — 您可能想在文件头中保存的可选纯文本元数据。例如,它可以用于指定有关底层张量的更多信息。这纯粹是信息性的,不影响张量加载。
将张量字典以 safetensors 格式保存为原始字节。
示例
from safetensors.flax import save_file
from jax import numpy as jnp
tensors = {"embedding": jnp.zeros((512, 1024)), "attention": jnp.zeros((256, 256))}
save_file(tensors, "model.safetensors")
safetensors.flax.save
< 源代码 > ( tensors: typing.Dict[str, jax.Array] metadata: typing.Optional[typing.Dict[str, str]] = None ) → bytes
参数
- tensors (
Dict[str, Array]
) — 待保存的张量。张量需要是连续且密集的。 - metadata (
Dict[str, str]
, 可选, 默认为 None
) — 您可能想在文件头中保存的可选纯文本元数据。例如,它可以用于指定有关底层张量的更多信息。这纯粹是信息性的,不影响张量加载。
代表该格式的原始字节
将张量字典以 safetensors 格式保存为原始字节。
示例
from safetensors.flax import save
from jax import numpy as jnp
tensors = {"embedding": jnp.zeros((512, 1024)), "attention": jnp.zeros((256, 256))}
byte_data = save(tensors)
< > 在 GitHub 上更新
© . This site is unofficial and not affiliated with Hugging Face, Inc.