Support both faiss-cpu and faiss-gpu in pyproject

vanna-ai · Sep 21, 2024 · f5bc5fa · f5bc5fa
1 parent 86ac7cd
commit f5bc5fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,4 +53,5 @@ milvus = ["pymilvus[model]"]
 bedrock = ["boto3", "botocore"]
 weaviate = ["weaviate-client"]
 azuresearch = ["azure-search-documents", "azure-identity", "azure-common", "fastembed"]
-faiss = ["faiss-cpu"]
+faiss-cpu = ["faiss-cpu"]
+faiss-gpu = ["faiss-gpu"]