From d13f44f48f3f059a3688005d040abe29cb3e834b Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Wed, 19 Jun 2024 17:17:01 -0400 Subject: [PATCH] We do not need per-version numpy anymore --- torchbenchmark/models/hf_Whisper/requirements.txt | 4 ---- torchbenchmark/models/tacotron2/requirements.txt | 5 +---- torchbenchmark/util/framework/detectron2/requirements.txt | 5 +---- 3 files changed, 2 insertions(+), 12 deletions(-) diff --git a/torchbenchmark/models/hf_Whisper/requirements.txt b/torchbenchmark/models/hf_Whisper/requirements.txt index 461b9ef776..c3db4451e8 100644 --- a/torchbenchmark/models/hf_Whisper/requirements.txt +++ b/torchbenchmark/models/hf_Whisper/requirements.txt @@ -1,5 +1 @@ numba -# We need to pin numpy version to the same as the torch testing environment -# which still supports python 3.8 -numpy==1.21.2; python_version < '3.11' -numpy==1.26.0; python_version >= '3.11' \ No newline at end of file diff --git a/torchbenchmark/models/tacotron2/requirements.txt b/torchbenchmark/models/tacotron2/requirements.txt index 4f642207d5..39e8afb167 100644 --- a/torchbenchmark/models/tacotron2/requirements.txt +++ b/torchbenchmark/models/tacotron2/requirements.txt @@ -1,7 +1,4 @@ -# We need to pin numpy version to the same as the torch testing environment -# which still supports python 3.8 -numpy==1.21.2; python_version < '3.11' -numpy==1.26.0; python_version >= '3.11' +numpy inflect scipy Unidecode diff --git a/torchbenchmark/util/framework/detectron2/requirements.txt b/torchbenchmark/util/framework/detectron2/requirements.txt index acec6a1840..f38075ddd7 100644 --- a/torchbenchmark/util/framework/detectron2/requirements.txt +++ b/torchbenchmark/util/framework/detectron2/requirements.txt @@ -1,6 +1,3 @@ git+https://github.com/facebookresearch/detectron2.git@0df2d73d0013db7de629602c23cc120219b4f2b8 omegaconf==2.3.0 -# We need to pin numpy version to the same as the torch testing environment -# which still supports python 3.8 -numpy==1.21.2; python_version < '3.11' -numpy==1.26.0; python_version >= '3.11' +numpy