From 29afbaaae3f1ab7f4fbff46e69754318883f30e1 Mon Sep 17 00:00:00 2001 From: NetBUG Date: Sat, 14 Sep 2024 17:14:40 +0200 Subject: [PATCH] Update dependencies to work with torch v2.2.x, 2.3.x etc --- requirements.txt | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/requirements.txt b/requirements.txt index 120f9085..703eeab9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,19 +2,21 @@ av==11.0.0 einops flashy>=0.0.1 +huggingface_hub hydra-core>=1.1 hydra_colorlog julius num2words -numpy<2.0.0 +numpy>=1.26.4,<2.0.0 sentencepiece spacy>=3.6.1 -torch==2.1.0 -torchaudio>=2.0.0,<2.1.2 -huggingface_hub +torch>=2.1.0,<2.5.0 +torchaudio>=2.0.0,<2.5.0 +torchvision>=0.16.0,<0.20.0 +torchtext>=0.16.0,<0.20.0 tqdm transformers>=4.31.0 # need Encodec there. -xformers<0.0.23 +xformers>=0.0.22,<0.0.28 demucs librosa soundfile @@ -22,7 +24,5 @@ gradio torchmetrics encodec protobuf -torchvision==0.16.0 -torchtext==0.16.0 pesq pystoi