Hugging Face's logo

Safetensors documentation

API de Tensorflow

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v0.5.0-rc.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

API de Tensorflow

safetensors.tensorflow.load_file

< >

( filename: typing.Union[str, os.PathLike]backend: str = 'mmap' ) → Dict[str, tf.Tensor]

Parámetros

  • filename (str, o os.PathLike) — El nombre del archivo que contiene los tensores
  • backend (str, opcional, valor predeterminado "mmap") — Backend de almacenamiento utilizado para servir los bytes de los tensores. "mmap" (predeterminado) y "pread" usan pread(2) para leer los bytes de los tensores.

Devuelve

Dict[str, tf.Tensor]

diccionario que contiene el nombre como clave y el valor como tf.Tensor

Carga un archivo safetensors en formato tensorflow.

Ejemplo:

from safetensors.tensorflow import load_file

file_path = "./my_folder/bert.safetensors"
loaded = load_file(file_path)

safetensors.tensorflow.load

< >

( data: bytes ) → Dict[str, tf.Tensor]

Parámetros

  • data (bytes) — El contenido de un archivo safetensors

Devuelve

Dict[str, tf.Tensor]

diccionario que contiene el nombre como clave y el valor como tf.Tensor en cpu

Carga un archivo safetensors en formato tensorflow a partir de bytes puros.

Ejemplo:

from safetensors.tensorflow import load

file_path = "./my_folder/bert.safetensors"
with open(file_path, "rb") as f:
    data = f.read()

loaded = load(data)

safetensors.tensorflow.save_file

< >

( tensors: typing.Dict[str, tensorflow.Tensor]filename: typing.Union[str, os.PathLike]metadata: typing.Optional[typing.Dict[str, str]] = None ) → None

Parámetros

  • tensors (Dict[str, tf.Tensor]) — Los tensores entrantes. Los tensores deben ser contiguos y densos.
  • filename (str, o os.PathLike)) — El nombre del archivo en el que se está guardando.
  • metadata (Dict[str, str], opcional, valor predeterminado None) — Metadatos opcionales de solo texto que puede desear guardar en su encabezado. Por ejemplo, puede ser útil para especificar más detalles sobre los tensores subyacentes. Esto es puramente informativo y no afecta la carga de tensores.

Devuelve

None

Guarda un diccionario de tensores en bytes crudos en el formato safetensors.

Ejemplo:

from safetensors.tensorflow import save_file
import tensorflow as tf

tensors = {"embedding": tf.zeros((512, 1024)), "attention": tf.zeros((256, 256))}
save_file(tensors, "model.safetensors")

safetensors.tensorflow.save

< >

( tensors: typing.Dict[str, tensorflow.Tensor]metadata: typing.Optional[typing.Dict[str, str]] = None ) → bytes

Parámetros

  • tensors (Dict[str, tf.Tensor]) — Los tensores entrantes. Los tensores deben ser contiguos y densos.
  • metadata (Dict[str, str], opcional, valor predeterminado None) — Metadatos opcionales de solo texto que puede desear guardar en su encabezado. Por ejemplo, puede ser útil para especificar más detalles sobre los tensores subyacentes. Esto es puramente informativo y no afecta la carga de tensores.

Devuelve

bytes

Los bytes crudos que representan el formato

Guarda un diccionario de tensores en bytes crudos en el formato safetensors.

Ejemplo:

from safetensors.tensorflow import save
import tensorflow as tf

tensors = {"embedding": tf.zeros((512, 1024)), "attention": tf.zeros((256, 256))}
byte_data = save(tensors)
Update on GitHub

←API de Torch API de PaddlePaddle→