Pin numpy version to 1.26.4 (#218)

* Pin numpy to 1.26.4 to avoid breaking incomaptibility of 2.0.0 with torch. * Update requirements txt with new numpy version.
idealo · Jul 3, 2024 · 4bbdfc0 · 4bbdfc0
1 parent 837d86a
commit 4bbdfc0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
+numpy==1.26.4
 torch
 torchvision
 Pillow>=9.0

diff --git 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',