Merge pull request #473 from facebookresearch/fix_numpy

Set constraint on numpy dependency
facebookresearch · Jun 19, 2024 · adf0b04 · adf0b04
2 parents dff1dae + c14a6ed
commit adf0b04
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/actions/audiocraft_build/action.yml b/.github/actions/audiocraft_build/action.yml
@@ -21,7 +21,7 @@ runs:
       python3 -m venv env
       .  env/bin/activate
       python -m pip install --upgrade pip
-      pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0
+      pip install 'numpy<2' torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0
       pip install xformers
       pip install -e '.[dev,wm]'
   - name: System Dependencies

diff --git a/requirements.txt b/requirements.txt
@@ -6,7 +6,7 @@ hydra-core>=1.1
 hydra_colorlog
 julius
 num2words
-numpy
+numpy<2.0.0
 sentencepiece
 spacy>=3.6.1
 torch==2.1.0