diff --git a/requirements.txt b/requirements.txt
index fd5c9252..4369f80f 100644
--- 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
index ffff258f..4607b8da 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,7 @@
         "flask-cors",
         "flask",
         "jinja2",
+        "safetensors",
         "torch",
         "transformers",
         "waitress",