Upgrade numpy to 2.0.0rc (#2311)

Summary: PyTorch upgraded to numpy 2.0.0rc1 which breaks a couple of models, we can fix all of them except DALLE2_pytorch. DALLE2_pytorch requires embeeding-reader, which requires numpy<2: https://github.com/rom1504/embedding-reader/blob/a4fd55830a502685600ed8ef07947cd1cb92b083/requirements.txt#L3 Therefore, we have to move DALLE2_pytorch to canary_models directory. Pytorch PR: pytorch/pytorch#128879 Pytorch SEV: pytorch/pytorch#128845 Pull Request resolved: #2311 Reviewed By: eellison Differential Revision: D58695403 Pulled By: xuzhao9 fbshipit-source-id: 052fa7c373089618686f99aec62c9ce4fd38e880
pytorch · Jun 17, 2024 · b949580 · b949580
1 parent 22999f6
commit b949580
Show file tree

Hide file tree

Showing 10 changed files with 6 additions and 8 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -15,9 +15,7 @@ transformers==4.38.1
 MonkeyType
 psutil
 pyyaml
-# pytorch build script pins numpy version
-# https://github.com/pytorch/builder/blob/main/wheel/build_wheel.sh
-numpy==1.23.5
+numpy
 opencv-python
 submitit
 pynvml
diff --git a/...nchmark/models/DALLE2_pytorch/__init__.py → .../canary_models/DALLE2_pytorch/__init__.py b/...nchmark/models/DALLE2_pytorch/__init__.py → .../canary_models/DALLE2_pytorch/__init__.py
diff --git a/...odels/DALLE2_pytorch/dalle2_pytorch.patch → ...odels/DALLE2_pytorch/dalle2_pytorch.patch b/...odels/DALLE2_pytorch/dalle2_pytorch.patch → ...odels/DALLE2_pytorch/dalle2_pytorch.patch
diff --git a/...enchmark/models/DALLE2_pytorch/install.py → ...k/canary_models/DALLE2_pytorch/install.py b/...enchmark/models/DALLE2_pytorch/install.py → ...k/canary_models/DALLE2_pytorch/install.py
diff --git a/...hmark/models/DALLE2_pytorch/metadata.yaml → ...anary_models/DALLE2_pytorch/metadata.yaml b/...hmark/models/DALLE2_pytorch/metadata.yaml → ...anary_models/DALLE2_pytorch/metadata.yaml
diff --git a/torchbenchmark/models/DALLE2_pytorch/origin → ...hmark/canary_models/DALLE2_pytorch/origin b/torchbenchmark/models/DALLE2_pytorch/origin → ...hmark/canary_models/DALLE2_pytorch/origin
diff --git a/...rk/models/DALLE2_pytorch/requirements.txt → ...ry_models/DALLE2_pytorch/requirements.txt b/...rk/models/DALLE2_pytorch/requirements.txt → ...ry_models/DALLE2_pytorch/requirements.txt
diff --git a/torchbenchmark/models/doctr_det_predictor/requirements.txt b/torchbenchmark/models/doctr_det_predictor/requirements.txt
@@ -1 +1 @@
-git+https://github.com/mindee/doctr.git@56c8356
+python-doctr
diff --git a/torchbenchmark/models/doctr_reco_predictor/requirements.txt b/torchbenchmark/models/doctr_reco_predictor/requirements.txt
@@ -1,2 +1,2 @@
-git+https://github.com/mindee/doctr.git@acb9f64
-rapidfuzz==2.15.1
+python-doctr
+rapidfuzz
diff --git a/torchbenchmark/models/timm_efficientdet/requirements.txt b/torchbenchmark/models/timm_efficientdet/requirements.txt
@@ -1,2 +1,2 @@
-pycocotools==2.0.6
-git+https://github.com/rwightman/efficientdet-pytorch.git@79d26d8
+pycocotools
+git+https://github.com/rwightman/efficientdet-pytorch.git@d43c9e34cd62d22b4205831bb735f6dd83b8e881