Hugging Face's logo

Safetensors documentation

واجهة برمجة تطبيقات 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

واجهة برمجة تطبيقات Tensorflow

safetensors.tensorflow.load_file

< >

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

المعاملات

  • filename (str، أو os.PathLike) — اسم الملف الذي يحتوي على المتجهات
  • backend (str، اختياري، القيمة الافتراضية "mmap") — خلفية التخزين المستخدمة لتقديم بايتات الموترات. يستخدم "mmap" (الافتراضي) و "pread" الأمر pread(2) لقراءة بايتات الموترات.

يعيد

Dict[str, tf.Tensor]

قاموس يحتوي على الاسم كمفتاح والقيمة كـtf.Tensor

يحمّل ملف safetensors إلى صيغة tensorflow.

مثال:

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]

المعاملات

  • data (bytes) — محتوى ملف safetensors

يعيد

Dict[str, tf.Tensor]

قاموس يحتوي على الاسم كمفتاح والقيمة كـtf.Tensor على وحدة المعالجة المركزية

يحمّل ملف safetensors إلى صيغة tensorflow من بايتات نقية.

مثال:

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

المعاملات

  • tensors (Dict[str, tf.Tensor]) — المصفوفات الواردة. يجب أن تكون المصفوفات متصلة وكثيفة.
  • filename (str أو os.PathLike) — اسم الملف الذي سنحفظ فيه.
  • metadata (Dict[str, str]، اختياري، القيمة الافتراضية None) — بيانات وصفية نصية اختيارية قد ترغب في حفظها في الترويسة. فعلى سبيل المثال، قد يكون من المفيد تحديد مزيد من التفاصيل حول المصفوفات الأساسية. هذا الأمر إخباري بحت ولا يؤثر على تحميل المصفوفات.

يعيد

None

يحفظ قاموسًا من المتجهات (tensors) في بايتات خام بصيغة safetensors.

مثال:

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

المعاملات

  • tensors (Dict[str, tf.Tensor]) — المصفوفات الواردة. يجب أن تكون المصفوفات متصلة وكثيفة.
  • metadata (Dict[str, str]، اختياري، القيمة الافتراضية None) — بيانات وصفية نصية اختيارية قد ترغب في حفظها في الترويسة. فعلى سبيل المثال، قد يكون من المفيد تحديد مزيد من التفاصيل حول المصفوفات الأساسية. هذا الأمر إخباري بحت ولا يؤثر على تحميل المصفوفات.

يعيد

bytes

البايتات الخام التي تمثل التنسيق

يحفظ قاموسًا من المتجهات (tensors) في بايتات خام بصيغة safetensors.

مثال:

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

←واجهة Torch البرمجية واجهة برمجة تطبيقات PaddlePaddle→