[Safetensors] make safetensors a required dep (#4177)

1e853e24 · Patrick von Platen · GitHub · ad787082 · 1e853e24 · 1e853e24
Unverified Commit 1e853e24 authored Jul 21, 2023 by Patrick von Platen Committed by GitHub Jul 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

setup.py setup.py +2 -1

src/diffusers/dependency_versions_table.py src/diffusers/dependency_versions_table.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -106,7 +106,7 @@ _deps = [
    "pytest-timeout",
    "pytest-xdist",
    "ruff>=0.0.241",
-    "safetensors",
+    "safetensors>=0.3.1",
    "sentencepiece>=0.1.91,!=0.1.92",
    "scipy",
    "onnx",
@@ -227,6 +227,7 @@ install_requires = [
    deps["numpy"],
    deps["regex"],
    deps["requests"],
+    deps["safetensors"],
    deps["Pillow"],
 ]


--- 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",
+    "safetensors": "safetensors>=0.3.1",
    "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
    "scipy": "scipy",
    "onnx": "onnx",