🚧 fix(wip): deps

Prismadic · Dec 23, 2023 · d6bf9d9 · d6bf9d9
1 parent 932047b
commit d6bf9d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/magnet/requirements.txt → requirements.txt b/magnet/requirements.txt → requirements.txt
@@ -3,7 +3,7 @@ sentence-transformers
 pandas
 tqdm
 spacy
-transformers
-faiss-cpu
+pymilvus
+mlx
 boto3
 nats-py
diff --git a/setup.py b/setup.py
@@ -12,6 +12,8 @@
         , "nats-py"
         , "sentence_transformers"
         , "vllm"
+        , "pymilvus"
+        , "mlx"
     ],
     url = 'https://github.com/Prismadic/magnet'
 )