upgrade safetensors version

vectorch-ai · Mar 1, 2025 · 0569c55 · 0569c55
1 parent 78fa907
commit 0569c55
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 14 deletions.
diff --git a/.github/workflows/build_wheel.yml b/.github/workflows/build_wheel.yml
@@ -24,7 +24,10 @@ jobs:
       matrix:
         python: ["3.9", "3.10", "3.11", "3.12"]
         cuda: ["11.8", "12.4", "12.6"]
-        torch: ["2.4.1", "2.5.1", "2.6.0"]
+        torch: ["2.5.1", "2.6.0"]
+        exclude:
+          - cuda: "12.6"
+            torch: "2.5.1"
     runs-on: [self-hosted, linux, release]
     env:
       PYTHON_VERSION: ${{ matrix.python }}

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -29,7 +29,6 @@ jobs:
       tag: ${{ needs.version-tag.outputs.tag }}
     secrets: inherit
     needs:
-      - version-tag
       - release-test
 
   # build wheel for release if tests pass
@@ -39,15 +38,14 @@ jobs:
       tag: ${{ needs.version-tag.outputs.tag }}
     secrets: inherit
     needs:
-      - version-tag
       - publish-wheel
 
   # publish docker image to dockerhub
-  publish-cpp-image:
-    uses: ./.github/workflows/publish_cpp_image.yml
-    with:
-      tag: ${{ needs.version-tag.outputs.tag }}
-    secrets: inherit
-    needs:
-      - version-tag
-      - publish-wheel
+  # publish-cpp-image:
+  #   uses: ./.github/workflows/publish_cpp_image.yml
+  #   with:
+  #     tag: ${{ needs.version-tag.outputs.tag }}
+  #   secrets: inherit
+  #   needs:
+  #     - version-tag
+  #     - publish-wheel
diff --git a/src/huggingface/Cargo.toml b/src/huggingface/Cargo.toml
@@ -11,8 +11,8 @@ crate-type = ["staticlib"]
 lto = "fat"
 
 [dependencies]
-thiserror = "1.0"
-safetensors = "0.4.4"
-tokenizers = { version = "0.20.0", default-features = false, features = ["onig"] }
+thiserror = "2.0"
+safetensors = "0.5.3"
+tokenizers = { version = "0.21.0", default-features = false, features = ["onig"] }
 serde = { version = "1.0", features = [ "derive" ] }
 serde_json = "1.0"