From fa74541bf3eabd405a802422a3aa103571470817 Mon Sep 17 00:00:00 2001
From: Gideon Giffard <118290024+fardeon@users.noreply.github.com>
Date: Fri, 21 Apr 2023 02:33:31 +0000
Subject: [PATCH] build(deps): add safetensors to dependencies

---
 requirements.txt | 1 +
 setup.py         | 1 +
 2 files changed, 2 insertions(+)

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",