safetensors 0.6.1rc0__tar.gz → 0.6.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: safetensors
-Version: 0.6.1rc0
+Version: 0.6.2
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Education

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

@@ -1,6 +1,6 @@
 # This file is automatically @generated by Cargo.
 # It is not intended for manual editing.
-version = 4
+version = 3
 
 [[package]]
 name = "autocfg"
@@ -40,9 +40,9 @@ checksum = "32a282da65faaf38286cf3be983213fcf1d2e2a58700e808f83f4ea9a4804bc0"
 
 [[package]]
 name = "memmap2"
-version = "0.9.7"
+version = "0.9.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "483758ad303d734cec05e5c12b41d7e93e6a6390c5e9dae6bdeb7c1259012d28"
+checksum = "fd3f7eed9d3848f8b98834af67102b720745c4ec028fcd0aa0239277e7de374f"
 dependencies = [
  "libc",
 ]
@@ -156,7 +156,7 @@ checksum = "28d3b2b1366ec20994f1fd18c3c594f05c5dd4bc44d8bb0c1c632c8d6829481f"
 
 [[package]]
 name = "safetensors"
-version = "0.6.1-rc.0"
+version = "0.6.2"
 dependencies = [
  "serde",
  "serde_json",
@@ -164,7 +164,7 @@ dependencies = [
 
 [[package]]
 name = "safetensors-python"
-version = "0.6.1-rc.0"
+version = "0.6.2"
 dependencies = [
  "memmap2",
  "pyo3",
@@ -194,9 +194,9 @@ dependencies = [
 
 [[package]]
 name = "serde_json"
-version = "1.0.142"
+version = "1.0.140"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "030fedb782600dcbd6f02d479bf0d817ac3bb40d644745b769d6a96bc3afc5a7"
+checksum = "20068b6e96dc6c9bd23e01df8827e6c7e1f2fddd43c21810382803c136b99373"
 dependencies = [
  "itoa",
  "memchr",

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

@@ -1,6 +1,6 @@
 [package]
 name = "safetensors-python"
-version = "0.6.1-rc.0"
+version = "0.6.2"
 edition = "2021"
 rust-version = "1.74"
 readme = "README.md"

{safetensors-0.6.1rc0 → safetensors-0.6.2/bindings/python}/py_src/safetensors/torch.py

@@ -433,7 +433,7 @@ _SIZE = {
     _float8_e8m0: 1,
     _float4_e2m1_x2: 1,
 }
-if Version(torch.__version__) > Version("2.0.0"):
+if Version(torch.__version__) >= Version("2.3.0"):
     _SIZE.update(
         {
             torch.uint64: 8,
@@ -456,7 +456,7 @@ _TYPES = {
     "F8_E4M3": _float8_e4m3fn,
     "F8_E5M2": _float8_e5m2,
 }
-if Version(torch.__version__) > Version("2.0.0"):
+if Version(torch.__version__) >= Version("2.3.0"):
     _TYPES.update(
         {
             "U64": torch.uint64,

{safetensors-0.6.1rc0 → safetensors-0.6.2}/bindings/python/src/lib.rs

@@ -33,7 +33,7 @@ struct PyView<'a> {
 }
 
 impl View for &PyView<'_> {
-    fn data(&self) -> std::borrow::Cow<[u8]> {
+    fn data(&self) -> std::borrow::Cow<'_, [u8]> {
         Cow::Borrowed(self.data.as_bytes())
     }
     fn shape(&self) -> &[usize] {

{safetensors-0.6.1rc0/bindings/python → safetensors-0.6.2}/py_src/safetensors/torch.py

@@ -433,7 +433,7 @@ _SIZE = {
     _float8_e8m0: 1,
     _float4_e2m1_x2: 1,
 }
-if Version(torch.__version__) > Version("2.0.0"):
+if Version(torch.__version__) >= Version("2.3.0"):
     _SIZE.update(
         {
             torch.uint64: 8,
@@ -456,7 +456,7 @@ _TYPES = {
     "F8_E4M3": _float8_e4m3fn,
     "F8_E5M2": _float8_e5m2,
 }
-if Version(torch.__version__) > Version("2.0.0"):
+if Version(torch.__version__) >= Version("2.3.0"):
     _TYPES.update(
         {
             "U64": torch.uint64,

{safetensors-0.6.1rc0 → safetensors-0.6.2}/safetensors/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "safetensors"
-version = "0.6.1-rc.0"
+version = "0.6.2"
 edition = "2021"
 rust-version = "1.80"
 homepage = "https://github.com/huggingface/safetensors"

{safetensors-0.6.1rc0 → safetensors-0.6.2}/safetensors/src/tensor.rs

@@ -142,7 +142,7 @@ struct PreparedData {
 ///    fn shape(&self) -> &[usize]{
 ///         &self.shape
 ///    }
-///    fn data(&self) -> Cow<[u8]>{
+///    fn data(&self) -> Cow<'_, [u8]>{
 ///        (&self.data).into()
 ///    }
 ///    fn data_len(&self) -> usize{
@@ -166,7 +166,7 @@ struct PreparedData {
 ///    fn shape(&self) -> &[usize]{
 ///         &self.shape
 ///    }
-///    fn data(&self) -> Cow<[u8]>{
+///    fn data(&self) -> Cow<'_, [u8]>{
 ///        self.data.into()
 ///    }
 ///    fn data_len(&self) -> usize{
@@ -193,7 +193,7 @@ struct PreparedData {
 ///    fn shape(&self) -> &[usize]{
 ///         &self.shape
 ///    }
-///    fn data(&self) -> Cow<[u8]>{
+///    fn data(&self) -> Cow<'_, [u8]>{
 ///        // This copies data from GPU to CPU.
 ///        let data: Vec<u8> = self.data.to_vec();
 ///        data.into()
@@ -211,7 +211,7 @@ pub trait View {
     /// The shape of the tensor
     fn shape(&self) -> &[usize];
     /// The data of the tensor
-    fn data(&self) -> Cow<[u8]>;
+    fn data(&self) -> Cow<'_, [u8]>;
     /// The length of the data, in bytes.
     /// This is necessary as this might be faster to get than `data().len()`
     /// for instance for tensors residing in GPU.
@@ -678,7 +678,7 @@ impl View for &TensorView<'_> {
         &self.shape
     }
 
-    fn data(&self) -> Cow<[u8]> {
+    fn data(&self) -> Cow<'_, [u8]> {
         self.data.into()
     }
 
@@ -696,7 +696,7 @@ impl View for TensorView<'_> {
         &self.shape
     }
 
-    fn data(&self) -> Cow<[u8]> {
+    fn data(&self) -> Cow<'_, [u8]> {
         self.data.into()
     }