diff --git a/PyTorch/built-in/audio/Stable_Audio_Tools/setup.py b/PyTorch/built-in/audio/Stable_Audio_Tools/setup.py index 86535a33f81d503f8639d38c09bc0cf834aef306..2b7ec615aa3a40bfc12411ab52246bec5d88b4f1 100644 --- a/PyTorch/built-in/audio/Stable_Audio_Tools/setup.py +++ b/PyTorch/built-in/audio/Stable_Audio_Tools/setup.py @@ -8,6 +8,7 @@ setup( description='Training and inference tools for generative audio models from Stability AI', packages=find_packages(), install_requires=[ + 'accelerate==0.33.0', 'aeiou==0.0.20', 'alias-free-torch==0.0.6', 'auraloss==0.4.0', @@ -34,7 +35,7 @@ setup( 'torchaudio==2.1.0', 'torchmetrics==0.11.4', 'tqdm', - 'transformers==4.51.3', + 'transformers==4.43.2', 'v-diffusion-pytorch==0.0.2', 'vector-quantize-pytorch==1.9.14', 'wandb==0.15.4',