build(deps): add safetensors to dependencies (hyperonym#175)

DataDropp · Apr 21, 2023 · 07fea19 · 07fea19
1 parent c99ef9f
commit 07fea19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,6 +8,7 @@ flask>=2.2.1
 huggingface-hub~=0.13.4
 jinja2>=3.1.2
 pytest>=6.2.5
+safetensors~=0.3.0
 torch>=1.12.1
 transformers[sentencepiece]~=4.28.1
 waitress~=2.1.2
diff --git a/setup.py b/setup.py
@@ -23,6 +23,7 @@
         "flask-cors",
         "flask",
         "jinja2",
+        "safetensors",
         "torch",
         "transformers",
         "waitress",