From 68d65f3e3c3c89294a99cc8e9c3144b26c2fabd0 Mon Sep 17 00:00:00 2001 From: tanujjain Date: Tue, 2 Jul 2024 18:32:02 +0200 Subject: [PATCH] Pin numpy to 1.26.4 to avoid breaking incomaptibility of 2.0.0 with torch. --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index e4b7c5c7..e5317d72 100644 --- a/setup.py +++ b/setup.py @@ -96,6 +96,7 @@ long_description=long_description, license='Apache 2.0', install_requires=[ + 'numpy==1.26.4', 'torch', 'torchvision', 'Pillow>=9.0',