diff --git a/.github/actions/audiocraft_build/action.yml b/.github/actions/audiocraft_build/action.yml index 74ab1e38..82835519 100644 --- 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 index effd8c7e..120f9085 100644 --- 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