Skip to content

GitLab

Unverified Commit 1e853e24 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

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

parent ad787082
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
setup.py
View file @ 1e853e24
...@@ -106,7 +106,7 @@ _deps = [ ...@@ -106,7 +106,7 @@ _deps = [
"pytest-timeout", "pytest-timeout",
"pytest-xdist", "pytest-xdist",
"ruff>=0.0.241", "ruff>=0.0.241",
"safetensors", "safetensors>=0.3.1",
"sentencepiece>=0.1.91,!=0.1.92", "sentencepiece>=0.1.91,!=0.1.92",
"scipy", "scipy",
"onnx", "onnx",
...@@ -227,6 +227,7 @@ install_requires = [ ...@@ -227,6 +227,7 @@ install_requires = [
deps["numpy"], deps["numpy"],
deps["regex"], deps["regex"],
deps["requests"], deps["requests"],
deps["safetensors"],
deps["Pillow"], deps["Pillow"],
] ]
......
src/diffusers/dependency_versions_table.py
View file @ 1e853e24
...@@ -30,7 +30,7 @@ deps = { ...@@ -30,7 +30,7 @@ deps = {
"pytest-timeout": "pytest-timeout", "pytest-timeout": "pytest-timeout",
"pytest-xdist": "pytest-xdist", "pytest-xdist": "pytest-xdist",
"ruff": "ruff>=0.0.241", "ruff": "ruff>=0.0.241",
"safetensors": "safetensors", "safetensors": "safetensors>=0.3.1",
"sentencepiece": "sentencepiece>=0.1.91,!=0.1.92", "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",
"scipy": "scipy", "scipy": "scipy",
"onnx": "onnx", "onnx": "onnx",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment