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

fardeon · web-flow · commit 07fea19972af · 2023-04-21T11:03:35.000+08:00
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",
