diff --git a/setup.py b/setup.py index fca2c1fccf..3a31361f1e 100644 --- a/setup.py +++ b/setup.py @@ -121,7 +121,7 @@ def localversion_func(version: ScmVersion) -> str: else "requests>=2.32.2" ), ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"), - ("torch>=2.9.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.9.0,<=2.9.1"), + ("torch>=2.9.0,<=2.10.0" if BUILD_TYPE == "release" else "torch>=2.9.0"), ( "transformers>=4.56.1,<=4.57.6" if BUILD_TYPE == "release" @@ -163,10 +163,10 @@ def localversion_func(version: ScmVersion) -> str: "cmarkgfm>=2024.1.14", "trl>=0.10.1", "pandas<2.3.0", - "torchvision<=0.24.1", + "torchvision", "librosa==0.11.0", "soundfile", - "torchcodec<=0.9.1", + "torchcodec", # linting, formatting, and type checking "mypy~=1.10.0", "ruff~=0.4.8",