[do not merge] safetensors pre release check

2025-12-07 04:54:47 +08:00 · 2023-08-07 16:56:52 +02:00
2 changed files with 2 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -106,7 +106,7 @@ _deps = [
    "pytest-timeout",
    "pytest-xdist",
    "ruff>=0.0.241",
-    "safetensors>=0.3.1",
+    "safetensors==0.3.2.rc1",
    "sentencepiece>=0.1.91,!=0.1.92",
    "scipy",
    "onnx",
--- a/src/diffusers/dependency_versions_table.py
+++ b/src/diffusers/dependency_versions_table.py
@@ -30,7 +30,7 @@ deps = {
    "pytest-timeout": "pytest-timeout",
    "pytest-xdist": "pytest-xdist",
    "ruff": "ruff>=0.0.241",
-    "safetensors": "safetensors>=0.3.1",
+    "safetensors": "safetensors==0.3.2.rc1",
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
    "scipy": "scipy",
    "onnx": "onnx",