autofaiss, haystack-ai, txtai: use faiss-cpu
This commit is contained in:
@@ -8,7 +8,7 @@
|
||||
|
||||
# dependencies
|
||||
embedding-reader,
|
||||
faiss,
|
||||
faiss-cpu,
|
||||
fire,
|
||||
fsspec,
|
||||
numpy,
|
||||
@@ -37,8 +37,6 @@ buildPythonPackage rec {
|
||||
# The `dataclasses` packages is a python2-only backport, unnecessary in
|
||||
# python3.
|
||||
"dataclasses"
|
||||
# We call it faiss, not faiss-cpu.
|
||||
"faiss-cpu"
|
||||
];
|
||||
|
||||
pythonRelaxDeps = [
|
||||
@@ -55,7 +53,7 @@ buildPythonPackage rec {
|
||||
|
||||
dependencies = [
|
||||
embedding-reader
|
||||
faiss
|
||||
faiss-cpu
|
||||
fire
|
||||
fsspec
|
||||
numpy
|
||||
|
||||
@@ -68,7 +68,7 @@
|
||||
seqeval,
|
||||
pdf2image,
|
||||
pytesseract,
|
||||
faiss,
|
||||
faiss-cpu,
|
||||
# , faiss-gpu
|
||||
pinecone-client,
|
||||
onnxruntime,
|
||||
@@ -105,11 +105,6 @@ buildPythonPackage rec {
|
||||
hatchling
|
||||
];
|
||||
|
||||
pythonRemoveDeps = [
|
||||
# We call it faiss, not faiss-cpu.
|
||||
"faiss-cpu"
|
||||
];
|
||||
|
||||
propagatedBuildInputs = [
|
||||
boilerpy3
|
||||
events
|
||||
@@ -210,7 +205,7 @@ buildPythonPackage rec {
|
||||
pdf2image
|
||||
pytesseract
|
||||
];
|
||||
only-faiss = [ faiss ];
|
||||
only-faiss = [ faiss-cpu ];
|
||||
# only-faiss-gpu = [
|
||||
# faiss-gpu
|
||||
# ];
|
||||
|
||||
@@ -7,7 +7,7 @@
|
||||
setuptools,
|
||||
|
||||
# dependencies
|
||||
faiss,
|
||||
faiss-cpu,
|
||||
torch,
|
||||
transformers,
|
||||
huggingface-hub,
|
||||
@@ -251,13 +251,8 @@ buildPythonPackage {
|
||||
|
||||
build-system = [ setuptools ];
|
||||
|
||||
pythonRemoveDeps = [
|
||||
# We call it faiss, not faiss-cpu.
|
||||
"faiss-cpu"
|
||||
];
|
||||
|
||||
dependencies = [
|
||||
faiss
|
||||
faiss-cpu
|
||||
huggingface-hub
|
||||
msgpack
|
||||
numpy
|
||||
|
||||
Reference in New Issue
Block a user