From f5bc5fa13b9cca65472db5dc7047c5c5be65c416 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= Date: Sat, 21 Sep 2024 11:34:52 +0200 Subject: [PATCH] Support both faiss-cpu and faiss-gpu in pyproject --- pyproject.toml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index a50febc8..7f439c1e 100644 --- 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"]