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.