safetensors 0.6.0rc0__tar.gz → 0.6.1rc0__tar.gz

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: safetensors
-Version: 0.6.0rc0
+Version: 0.6.1rc0
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Education
@@ -28,10 +28,7 @@ Requires-Dist: jaxlib>=0.3.25 ; extra == 'jax'
 Requires-Dist: mlx>=0.0.9 ; extra == 'mlx'
 Requires-Dist: safetensors[numpy] ; extra == 'paddlepaddle'
 Requires-Dist: paddlepaddle>=2.4.1 ; extra == 'paddlepaddle'
-Requires-Dist: black==22.3 ; extra == 'quality'
-Requires-Dist: click==8.0.4 ; extra == 'quality'
-Requires-Dist: isort>=5.5.4 ; extra == 'quality'
-Requires-Dist: flake8>=3.8.3 ; extra == 'quality'
+Requires-Dist: ruff ; extra == 'quality'
 Requires-Dist: safetensors[numpy] ; extra == 'testing'
 Requires-Dist: h5py>=3.7.0 ; extra == 'testing'
 Requires-Dist: huggingface-hub>=0.12.1 ; extra == 'testing'
@@ -39,6 +36,12 @@ Requires-Dist: setuptools-rust>=1.5.2 ; extra == 'testing'
 Requires-Dist: pytest>=7.2.0 ; extra == 'testing'
 Requires-Dist: pytest-benchmark>=4.0.0 ; extra == 'testing'
 Requires-Dist: hypothesis>=6.70.2 ; extra == 'testing'
+Requires-Dist: safetensors[numpy] ; extra == 'testingfree'
+Requires-Dist: huggingface-hub>=0.12.1 ; extra == 'testingfree'
+Requires-Dist: setuptools-rust>=1.5.2 ; extra == 'testingfree'
+Requires-Dist: pytest>=7.2.0 ; extra == 'testingfree'
+Requires-Dist: pytest-benchmark>=4.0.0 ; extra == 'testingfree'
+Requires-Dist: hypothesis>=6.70.2 ; extra == 'testingfree'
 Requires-Dist: safetensors[torch] ; extra == 'all'
 Requires-Dist: safetensors[numpy] ; extra == 'all'
 Requires-Dist: safetensors[pinned-tf] ; extra == 'all'
@@ -56,6 +59,7 @@ Provides-Extra: mlx
 Provides-Extra: paddlepaddle
 Provides-Extra: quality
 Provides-Extra: testing
+Provides-Extra: testingfree
 Provides-Extra: all
 Provides-Extra: dev
 License-File: LICENSE

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/Cargo.lock

@@ -1,12 +1,12 @@
 # This file is automatically @generated by Cargo.
 # It is not intended for manual editing.
-version = 3
+version = 4
 
 [[package]]
 name = "autocfg"
-version = "1.4.0"
+version = "1.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26"
+checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
 
 [[package]]
 name = "heck"
@@ -28,9 +28,9 @@ checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
 
 [[package]]
 name = "libc"
-version = "0.2.173"
+version = "0.2.174"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d8cfeafaffdbc32176b64fb251369d52ea9f0a8fbc6f8759edffef7b525d64bb"
+checksum = "1171693293099992e19cddea4e8b849964e9846f4acee11b3948bcc337be8776"
 
 [[package]]
 name = "memchr"
@@ -40,9 +40,9 @@ checksum = "32a282da65faaf38286cf3be983213fcf1d2e2a58700e808f83f4ea9a4804bc0"
 
 [[package]]
 name = "memmap2"
-version = "0.9.5"
+version = "0.9.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "fd3f7eed9d3848f8b98834af67102b720745c4ec028fcd0aa0239277e7de374f"
+checksum = "483758ad303d734cec05e5c12b41d7e93e6a6390c5e9dae6bdeb7c1259012d28"
 dependencies = [
  "libc",
 ]
@@ -156,7 +156,7 @@ checksum = "28d3b2b1366ec20994f1fd18c3c594f05c5dd4bc44d8bb0c1c632c8d6829481f"
 
 [[package]]
 name = "safetensors"
-version = "0.6.0-rc.0"
+version = "0.6.1-rc.0"
 dependencies = [
  "serde",
  "serde_json",
@@ -164,7 +164,7 @@ dependencies = [
 
 [[package]]
 name = "safetensors-python"
-version = "0.6.0-rc.0"
+version = "0.6.1-rc.0"
 dependencies = [
  "memmap2",
  "pyo3",
@@ -194,9 +194,9 @@ dependencies = [
 
 [[package]]
 name = "serde_json"
-version = "1.0.140"
+version = "1.0.142"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "20068b6e96dc6c9bd23e01df8827e6c7e1f2fddd43c21810382803c136b99373"
+checksum = "030fedb782600dcbd6f02d479bf0d817ac3bb40d644745b769d6a96bc3afc5a7"
 dependencies = [
  "itoa",
  "memchr",
@@ -206,9 +206,9 @@ dependencies = [
 
 [[package]]
 name = "syn"
-version = "2.0.103"
+version = "2.0.104"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "e4307e30089d6fd6aff212f2da3a1f9e32f3223b1f010fb09b7c95f90f3ca1e8"
+checksum = "17b6f705963418cdb9927482fa304bc562ece2fdd4f616084c50b7023b435a40"
 dependencies = [
  "proc-macro2",
  "quote",

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/Cargo.toml

@@ -1,8 +1,9 @@
 [package]
 name = "safetensors-python"
-version = "0.6.0-rc.0"
+version = "0.6.1-rc.0"
 edition = "2021"
 rust-version = "1.74"
+readme = "README.md"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [lib]

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/benches/test_pt.py

@@ -118,7 +118,10 @@ def test_pt_sf_load_gpu(benchmark):
         assert torch.allclose(v, tv)
 
 
-@pytest.mark.skipif(not hasattr(torch.backends, "mps") or not torch.backends.mps.is_available(), reason="requires mps")
+@pytest.mark.skipif(
+    not hasattr(torch.backends, "mps") or not torch.backends.mps.is_available(),
+    reason="requires mps",
+)
 def test_pt_pt_load_mps(benchmark):
     # benchmark something
     weights = create_gpt2(12)
@@ -133,7 +136,10 @@ def test_pt_pt_load_mps(benchmark):
         assert torch.allclose(v, tv)
 
 
-@pytest.mark.skipif(not hasattr(torch.backends, "mps") or not torch.backends.mps.is_available(), reason="requires mps")
+@pytest.mark.skipif(
+    not hasattr(torch.backends, "mps") or not torch.backends.mps.is_available(),
+    reason="requires mps",
+)
 def test_pt_sf_load_mps(benchmark):
     # benchmark something
     weights = create_gpt2(12)

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/convert.py

@@ -8,7 +8,13 @@ from typing import Dict, List, Optional, Set, Tuple
 
 import torch
 
-from huggingface_hub import CommitInfo, CommitOperationAdd, Discussion, HfApi, hf_hub_download
+from huggingface_hub import (
+    CommitInfo,
+    CommitOperationAdd,
+    Discussion,
+    HfApi,
+    hf_hub_download,
+)
 from huggingface_hub.file_download import repo_folder_name
 from safetensors.torch import _find_shared_tensors, _is_complete, load_file, save_file
 
@@ -49,7 +55,9 @@ def _remove_duplicate_names(
     shareds = _find_shared_tensors(state_dict)
     to_remove = defaultdict(list)
     for shared in shareds:
-        complete_names = set([name for name in shared if _is_complete(state_dict[name])])
+        complete_names = set(
+            [name for name in shared if _is_complete(state_dict[name])]
+        )
         if not complete_names:
             if len(shared) == 1:
                 # Force contiguous
@@ -81,14 +89,20 @@ def _remove_duplicate_names(
     return to_remove
 
 
-def get_discard_names(model_id: str, revision: Optional[str], folder: str, token: Optional[str]) -> List[str]:
+def get_discard_names(
+    model_id: str, revision: Optional[str], folder: str, token: Optional[str]
+) -> List[str]:
     try:
         import json
 
         import transformers
 
         config_filename = hf_hub_download(
-            model_id, revision=revision, filename="config.json", token=token, cache_dir=folder
+            model_id,
+            revision=revision,
+            filename="config.json",
+            token=token,
+            cache_dir=folder,
         )
         with open(config_filename, "r") as f:
             config = json.load(f)
@@ -129,10 +143,19 @@ def rename(pt_filename: str) -> str:
 
 
 def convert_multi(
-    model_id: str, *, revision=Optional[str], folder: str, token: Optional[str], discard_names: List[str]
+    model_id: str,
+    *,
+    revision=Optional[str],
+    folder: str,
+    token: Optional[str],
+    discard_names: List[str],
 ) -> ConversionResult:
     filename = hf_hub_download(
-        repo_id=model_id, revision=revision, filename="pytorch_model.bin.index.json", token=token, cache_dir=folder
+        repo_id=model_id,
+        revision=revision,
+        filename="pytorch_model.bin.index.json",
+        token=token,
+        cache_dir=folder,
     )
     with open(filename, "r") as f:
         data = json.load(f)
@@ -140,7 +163,9 @@ def convert_multi(
     filenames = set(data["weight_map"].values())
     local_filenames = []
     for filename in filenames:
-        pt_filename = hf_hub_download(repo_id=model_id, filename=filename, token=token, cache_dir=folder)
+        pt_filename = hf_hub_download(
+            repo_id=model_id, filename=filename, token=token, cache_dir=folder
+        )
 
         sf_filename = rename(pt_filename)
         sf_filename = os.path.join(folder, sf_filename)
@@ -156,7 +181,8 @@ def convert_multi(
     local_filenames.append(index)
 
     operations = [
-        CommitOperationAdd(path_in_repo=os.path.basename(local), path_or_fileobj=local) for local in local_filenames
+        CommitOperationAdd(path_in_repo=os.path.basename(local), path_or_fileobj=local)
+        for local in local_filenames
     ]
     errors: List[Tuple[str, "Exception"]] = []
 
@@ -164,10 +190,19 @@ def convert_multi(
 
 
 def convert_single(
-    model_id: str, *, revision: Optional[str], folder: str, token: Optional[str], discard_names: List[str]
+    model_id: str,
+    *,
+    revision: Optional[str],
+    folder: str,
+    token: Optional[str],
+    discard_names: List[str],
 ) -> ConversionResult:
     pt_filename = hf_hub_download(
-        repo_id=model_id, revision=revision, filename="pytorch_model.bin", token=token, cache_dir=folder
+        repo_id=model_id,
+        revision=revision,
+        filename="pytorch_model.bin",
+        token=token,
+        cache_dir=folder,
     )
 
     sf_name = "model.safetensors"
@@ -219,20 +254,30 @@ def create_diff(pt_infos: Dict[str, List[str]], sf_infos: Dict[str, List[str]])
         sf_only = sf_set - pt_set
 
         if pt_only:
-            errors.append(f"{key} : PT warnings contain {pt_only} which are not present in SF warnings")
+            errors.append(
+                f"{key} : PT warnings contain {pt_only} which are not present in SF warnings"
+            )
         if sf_only:
-            errors.append(f"{key} : SF warnings contain {sf_only} which are not present in PT warnings")
+            errors.append(
+                f"{key} : SF warnings contain {sf_only} which are not present in PT warnings"
+            )
     return "\n".join(errors)
 
 
-def previous_pr(api: "HfApi", model_id: str, pr_title: str, revision=Optional[str]) -> Optional["Discussion"]:
+def previous_pr(
+    api: "HfApi", model_id: str, pr_title: str, revision=Optional[str]
+) -> Optional["Discussion"]:
     try:
         revision_commit = api.model_info(model_id, revision=revision).sha
         discussions = api.get_repo_discussions(repo_id=model_id)
     except Exception:
         return None
     for discussion in discussions:
-        if discussion.status in {"open", "closed"} and discussion.is_pull_request and discussion.title == pr_title:
+        if (
+            discussion.status in {"open", "closed"}
+            and discussion.is_pull_request
+            and discussion.title == pr_title
+        ):
             commits = api.list_repo_commits(model_id, revision=discussion.git_reference)
 
             if revision_commit == commits[1].commit_id:
@@ -241,7 +286,12 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str, revision=Optional[st
 
 
 def convert_generic(
-    model_id: str, *, revision=Optional[str], folder: str, filenames: Set[str], token: Optional[str]
+    model_id: str,
+    *,
+    revision=Optional[str],
+    folder: str,
+    filenames: Set[str],
+    token: Optional[str],
 ) -> ConversionResult:
     operations = []
     errors = []
@@ -251,7 +301,11 @@ def convert_generic(
         prefix, ext = os.path.splitext(filename)
         if ext in extensions:
             pt_filename = hf_hub_download(
-                model_id, revision=revision, filename=filename, token=token, cache_dir=folder
+                model_id,
+                revision=revision,
+                filename=filename,
+                token=token,
+                cache_dir=folder,
             )
             dirname, raw_filename = os.path.split(filename)
             if raw_filename == "pytorch_model.bin":
@@ -263,7 +317,11 @@ def convert_generic(
             sf_filename = os.path.join(folder, sf_in_repo)
             try:
                 convert_file(pt_filename, sf_filename, discard_names=[])
-                operations.append(CommitOperationAdd(path_in_repo=sf_in_repo, path_or_fileobj=sf_filename))
+                operations.append(
+                    CommitOperationAdd(
+                        path_in_repo=sf_in_repo, path_or_fileobj=sf_filename
+                    )
+                )
             except Exception as e:
                 errors.append((pt_filename, e))
     return operations, errors
@@ -285,28 +343,50 @@ def convert(
             pr = previous_pr(api, model_id, pr_title, revision=revision)
 
             library_name = getattr(info, "library_name", None)
-            if any(filename.endswith(".safetensors") for filename in filenames) and not force:
-                raise AlreadyExists(f"Model {model_id} is already converted, skipping..")
+            if (
+                any(filename.endswith(".safetensors") for filename in filenames)
+                and not force
+            ):
+                raise AlreadyExists(
+                    f"Model {model_id} is already converted, skipping.."
+                )
             elif pr is not None and not force:
                 url = f"https://huggingface.co/{model_id}/discussions/{pr.num}"
                 new_pr = pr
-                raise AlreadyExists(f"Model {model_id} already has an open PR check out {url}")
+                raise AlreadyExists(
+                    f"Model {model_id} already has an open PR check out {url}"
+                )
             elif library_name == "transformers":
-
-                discard_names = get_discard_names(model_id, revision=revision, folder=folder, token=api.token)
+                discard_names = get_discard_names(
+                    model_id, revision=revision, folder=folder, token=api.token
+                )
                 if "pytorch_model.bin" in filenames:
                     operations, errors = convert_single(
-                        model_id, revision=revision, folder=folder, token=api.token, discard_names=discard_names
+                        model_id,
+                        revision=revision,
+                        folder=folder,
+                        token=api.token,
+                        discard_names=discard_names,
                     )
                 elif "pytorch_model.bin.index.json" in filenames:
                     operations, errors = convert_multi(
-                        model_id, revision=revision, folder=folder, token=api.token, discard_names=discard_names
+                        model_id,
+                        revision=revision,
+                        folder=folder,
+                        token=api.token,
+                        discard_names=discard_names,
                     )
                 else:
-                    raise RuntimeError(f"Model {model_id} doesn't seem to be a valid pytorch model. Cannot convert")
+                    raise RuntimeError(
+                        f"Model {model_id} doesn't seem to be a valid pytorch model. Cannot convert"
+                    )
             else:
                 operations, errors = convert_generic(
-                    model_id, revision=revision, folder=folder, filenames=filenames, token=api.token
+                    model_id,
+                    revision=revision,
+                    folder=folder,
+                    filenames=filenames,
+                    token=api.token,
                 )
 
             if operations:
@@ -366,7 +446,9 @@ if __name__ == "__main__":
             " Continue [Y/n] ?"
         )
     if txt.lower() in {"", "y"}:
-        commit_info, errors = convert(api, model_id, revision=args.revision, force=args.force)
+        commit_info, errors = convert(
+            api, model_id, revision=args.revision, force=args.force
+        )
         string = f"""
 ### Success 🔥
 Yay! This model was successfully converted and a PR was open using your token, here:
@@ -375,7 +457,8 @@ Yay! This model was successfully converted and a PR was open using your token, h
         if errors:
             string += "\nErrors during conversion:\n"
             string += "\n".join(
-                f"Error while converting {filename}: {e}, skipped conversion" for filename, e in errors
+                f"Error while converting {filename}: {e}, skipped conversion"
+                for filename, e in errors
             )
         print(string)
     else:

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/convert_all.py

@@ -1,4 +1,5 @@
 """Simple utility tool to convert automatically most downloaded models"""
+
 from convert import AlreadyExists, convert
 from huggingface_hub import HfApi, ModelFilter, ModelSearchArguments
 from transformers import AutoConfig
@@ -10,7 +11,11 @@ if __name__ == "__main__":
 
     total = 50
     models = list(
-        api.list_models(filter=ModelFilter(library=args.library.Transformers), sort="downloads", direction=-1)
+        api.list_models(
+            filter=ModelFilter(library=args.library.Transformers),
+            sort="downloads",
+            direction=-1,
+        )
     )[:total]
 
     correct = 0
@@ -40,4 +45,4 @@ if __name__ == "__main__":
 
     print(f"Errors: {errors}")
     print(f"File size is difference {len(errors)}")
-    print(f"Correct rate {correct}/{total} ({correct/total * 100:.2f}%)")
+    print(f"Correct rate {correct}/{total} ({correct / total * 100:.2f}%)")

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/py_src/safetensors/__init__.py

@@ -4,6 +4,7 @@ from ._safetensors_rust import (  # noqa: F401
     __version__,
     deserialize,
     safe_open,
+    _safe_open_handle,
     serialize,
     serialize_file,
 )

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/py_src/safetensors/__init__.pyi

@@ -49,7 +49,7 @@ def serialize_file(tensor_dict, filename, metadata=None):
 
     Returns:
         (`NoneType`):
-            On success return None.
+            On success return None
     """
     pass
 
@@ -68,19 +68,21 @@ class safe_open:
         device (`str`, defaults to `"cpu"`):
             The device on which you want the tensors.
     """
-
     def __init__(self, filename, framework, device=...):
         pass
+
     def __enter__(self):
         """
         Start the context manager
         """
         pass
+
     def __exit__(self, _exc_type, _exc_value, _traceback):
         """
         Exits the context manager
         """
         pass
+
     def get_slice(self, name):
         """
         Returns a full slice view object
@@ -102,6 +104,7 @@ class safe_open:
         ```
         """
         pass
+
     def get_tensor(self, name):
         """
         Returns a full tensor
@@ -124,6 +127,7 @@ class safe_open:
         ```
         """
         pass
+
     def keys(self):
         """
         Returns the names of the tensors in the file.
@@ -133,6 +137,7 @@ class safe_open:
                 The name of the tensors contained in that file
         """
         pass
+
     def metadata(self):
         """
         Return the special non tensor information in the header
@@ -143,6 +148,16 @@ class safe_open:
         """
         pass
 
+    def offset_keys(self):
+        """
+        Returns the names of the tensors in the file, ordered by offset.
+
+        Returns:
+            (`List[str]`):
+                The name of the tensors contained in that file
+        """
+        pass
+
 class SafetensorError(Exception):
     """
     Custom Python Exception for Safetensor errors.

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/py_src/safetensors/mlx.py

@@ -7,7 +7,9 @@ import mlx.core as mx
 from safetensors import numpy, safe_open
 
 
-def save(tensors: Dict[str, mx.array], metadata: Optional[Dict[str, str]] = None) -> bytes:
+def save(
+    tensors: Dict[str, mx.array], metadata: Optional[Dict[str, str]] = None
+) -> bytes:
     """
     Saves a dictionary of tensors into raw bytes in safetensors format.
 

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0}/bindings/python/py_src/safetensors/numpy.py

@@ -13,7 +13,9 @@ def _tobytes(tensor: np.ndarray) -> bytes:
     return tensor.tobytes()
 
 
-def save(tensor_dict: Dict[str, np.ndarray], metadata: Optional[Dict[str, str]] = None) -> bytes:
+def save(
+    tensor_dict: Dict[str, np.ndarray], metadata: Optional[Dict[str, str]] = None
+) -> bytes:
     """
     Saves a dictionary of tensors into raw bytes in safetensors format.
 
@@ -38,14 +40,19 @@ def save(tensor_dict: Dict[str, np.ndarray], metadata: Optional[Dict[str, str]]
     byte_data = save(tensors)
     ```
     """
-    flattened = {k: {"dtype": v.dtype.name, "shape": v.shape, "data": _tobytes(v)} for k, v in tensor_dict.items()}
+    flattened = {
+        k: {"dtype": v.dtype.name, "shape": v.shape, "data": _tobytes(v)}
+        for k, v in tensor_dict.items()
+    }
     serialized = serialize(flattened, metadata=metadata)
     result = bytes(serialized)
     return result
 
 
 def save_file(
-    tensor_dict: Dict[str, np.ndarray], filename: Union[str, os.PathLike], metadata: Optional[Dict[str, str]] = None
+    tensor_dict: Dict[str, np.ndarray],
+    filename: Union[str, os.PathLike],
+    metadata: Optional[Dict[str, str]] = None,
 ) -> None:
     """
     Saves a dictionary of tensors into raw bytes in safetensors format.
@@ -73,7 +80,10 @@ def save_file(
     save_file(tensors, "model.safetensors")
     ```
     """
-    flattened = {k: {"dtype": v.dtype.name, "shape": v.shape, "data": _tobytes(v)} for k, v in tensor_dict.items()}
+    flattened = {
+        k: {"dtype": v.dtype.name, "shape": v.shape, "data": _tobytes(v)}
+        for k, v in tensor_dict.items()
+    }
     serialize_file(flattened, filename, metadata=metadata)
 
 

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0/bindings/python}/py_src/safetensors/paddle.py

@@ -7,7 +7,9 @@ import paddle
 from safetensors import numpy
 
 
-def save(tensors: Dict[str, paddle.Tensor], metadata: Optional[Dict[str, str]] = None) -> bytes:
+def save(
+    tensors: Dict[str, paddle.Tensor], metadata: Optional[Dict[str, str]] = None
+) -> bytes:
     """
     Saves a dictionary of tensors into raw bytes in safetensors format.
 
@@ -98,7 +100,9 @@ def load(data: bytes, device: str = "cpu") -> Dict[str, paddle.Tensor]:
     return _np2paddle(flat, device)
 
 
-def load_file(filename: Union[str, os.PathLike], device="cpu") -> Dict[str, paddle.Tensor]:
+def load_file(
+    filename: Union[str, os.PathLike], device="cpu"
+) -> Dict[str, paddle.Tensor]:
     """
     Loads a safetensors file into paddle format.
 
@@ -126,7 +130,9 @@ def load_file(filename: Union[str, os.PathLike], device="cpu") -> Dict[str, padd
     return output
 
 
-def _np2paddle(numpy_dict: Dict[str, np.ndarray], device: str = "cpu") -> Dict[str, paddle.Tensor]:
+def _np2paddle(
+    numpy_dict: Dict[str, np.ndarray], device: str = "cpu"
+) -> Dict[str, paddle.Tensor]:
     for k, v in numpy_dict.items():
         numpy_dict[k] = paddle.to_tensor(v, place=device)
     return numpy_dict

{safetensors-0.6.0rc0 → safetensors-0.6.1rc0/bindings/python}/py_src/safetensors/tensorflow.py

@@ -7,7 +7,9 @@ import tensorflow as tf
 from safetensors import numpy, safe_open
 
 
-def save(tensors: Dict[str, tf.Tensor], metadata: Optional[Dict[str, str]] = None) -> bytes:
+def save(
+    tensors: Dict[str, tf.Tensor], metadata: Optional[Dict[str, str]] = None
+) -> bytes:
     """
     Saves a dictionary of tensors into raw bytes in safetensors format.