Pin numpy version

pytorch · Jun 19, 2024 · 23512db · 23512db
1 parent 42bb752
commit 23512db
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -15,7 +15,7 @@ transformers==4.38.1
 MonkeyType
 psutil
 pyyaml
-numpy
+numpy==1.21.2
 opencv-python
 submitit
 pynvml
diff --git a/torchbenchmark/models/Background_Matting/requirements.txt b/torchbenchmark/models/Background_Matting/requirements.txt
@@ -1,5 +1,5 @@
-numpy
 opencv-python
+numpy==1.21.2
 pandas
 Pillow
 scikit-image

diff --git a/torchbenchmark/models/hf_Whisper/requirements.txt b/torchbenchmark/models/hf_Whisper/requirements.txt
@@ -1 +1,2 @@
-numba
+numba
+numpy==1.21.2
diff --git a/torchbenchmark/models/tacotron2/requirements.txt b/torchbenchmark/models/tacotron2/requirements.txt
@@ -1,4 +1,4 @@
-numpy
+numpy==1.21.2
 inflect
 scipy
 Unidecode

diff --git a/torchbenchmark/util/framework/detectron2/requirements.txt b/torchbenchmark/util/framework/detectron2/requirements.txt
@@ -1,3 +1,3 @@
 git+https://github.com/facebookresearch/detectron2.git@1a4df4d
 omegaconf==2.3.0
-numpy
+numpy==1.21.2