diff --git a/setup.py b/setup.py index d893f9d..43b1727 100644 --- a/setup.py +++ b/setup.py @@ -13,7 +13,7 @@ python_requires='>=3.9', install_requires=['packaging', 'wheel', - 'torch==2.4', + 'torch==2.6.0', 'torchaudio==2.4', 'stable-audio-tools==0.0.17', 'pytorch-lightning==2.1',