[WIP] Test safetensors 0.3.3.rc1 sanity check

2025-12-10 14:34:55 +08:00 · 2023-08-23 13:51:02 +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.280",
-    "safetensors>=0.3.1",
+    "safetensors==0.3.3.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.280",
-    "safetensors": "safetensors>=0.3.1",
+    "safetensors": "safetensors==0.3.3.rc1",
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
    "scipy": "scipy",
    "onnx": "onnx",