fix: dependencies

breezedeus · Dec 6, 2023 · 2f37db6 · 2f37db6
1 parent 85fd88a
commit 2f37db6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -37,9 +37,13 @@
 )
 
 required = [
-    "click",
-    "tqdm",
     "colorlog",
+    "click",
+    "numpy",
+    "torch",
+    "torchvision",
+    "transformers",
+    "pillow",
     "chromadb",
 ]
 extras_require = {