From 98e35a5e96069951ef27e5812fb354df02f01c5c Mon Sep 17 00:00:00 2001 From: Jonatan Hanssen Date: Thu, 22 Aug 2024 11:38:41 +0200 Subject: [PATCH] Downgrade numpy to fix faiss dependency Signed-off-by: Jonatan Hanssen --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index e046f7e4..e63ad005 100644 --- a/setup.py +++ b/setup.py @@ -16,6 +16,7 @@ packages=setuptools.find_packages(), install_requires=[ 'torch>=1.13.1', + 'numpy<2', 'torchvision>=0.13', 'scikit-learn', 'json5',