chore: add some packages to requirments.txt

nuffin · Oct 18, 2024 · 747b944 · 747b944
1 parent 3a38f51
commit 747b944
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.4.1
+torch==2.4.1 ## for fastai
 torchaudio==2.4.1
 torchvision==0.19.1
 python-magic==0.4.27
@@ -17,5 +17,11 @@ black==24.10.0
 bcrypt==4.2.0
 uvicorn==0.32.0
 asgiref==3.8.1
-tensorflow==2.17.0
+tensorflow[and-cuda]==2.17.0
 transformers==4.45.2
+scikit-learn==1.5.2
+fastai==2.7.17
+# bark==0.1.5
+joblib==1.4.2
+dspy-ai==2.5.12
+pgvector==0.3.5