Pin numpy to 1.26.4 to avoid breaking incomaptibility of 2.0.0 with t…

…orch.
idealo · Jul 2, 2024 · 68d65f3 · 68d65f3
1 parent 837d86a
commit 68d65f3
Showing 1 changed file with 1 addition and 0 deletions.
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',