diff --git a/pyproject.toml b/pyproject.toml
index 14ed09dc..78d8b7ae 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -45,6 +45,7 @@ PyPDF2 = "*"
 accelerate = "*"
 sentencepiece = "*"
 wget = "*"
+tensorflow = "*"
 httpx = "*"
 tiktoken = "*"
 safetensors = "*"
diff --git a/requirements.txt b/requirements.txt
index 00824a1d..d256291e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,6 +25,7 @@ sentencepiece
 PyPDF2
 accelerate
 chromadb
+tensorflow
 optimum
 tiktoken
 tabulate