diff --git a/setup.py b/setup.py index 79325409c..7bdb31ffb 100644 --- a/setup.py +++ b/setup.py @@ -114,22 +114,24 @@ def localversion_func(version: ScmVersion) -> str: ("pyyaml>=6.0.1,<=6.0.3" if BUILD_TYPE == "release" else "pyyaml>=6.0.1"), # librosa dependency numba is currently not compatible with numpy>=2.3 # https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information - ("numpy>=2.0.0,<=2.3.3" if BUILD_TYPE == "release" else "numpy>=2.0.0"), + ("numpy>=2.0.0,<=2.3.5" if BUILD_TYPE == "release" else "numpy>=2.0.0"), ( "requests>=2.32.2,<=2.32.5" if BUILD_TYPE == "release" else "requests>=2.32.2" ), ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3"), - ("torch>=2.7.0,<=2.8.0" if BUILD_TYPE == "release" else "torch>=2.7.0"), + ("torch>=2.7.0,<=2.9.1" if BUILD_TYPE == "release" else "torch>=2.7.0"), ( - "transformers>=4.54.0,<=4.56.2" + "transformers>=4.54.0,<=4.57.3" if BUILD_TYPE == "release" else "transformers>=4.54.0" ), - ("datasets>=4.0.0,<=4.1.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), + ("datasets>=4.0.0,<=4.4.1" if BUILD_TYPE == "release" else "datasets>=4.0.0"), + ("auto-round==0.9.1" if BUILD_TYPE == "release" else "auto-round==0.9.1"), + # auto-round 0.9.1 requires accelerate >= 1.10.0 ( - "accelerate>=1.6.0,<=1.10.1" + "accelerate>=1.10.0,<=1.12.0" if BUILD_TYPE == "release" else "accelerate>=1.6.0" ), @@ -138,13 +140,12 @@ def localversion_func(version: ScmVersion) -> str: if BUILD_TYPE == "release" else "nvidia-ml-py>=12.560.30" ), - ("pillow>=10.4.0,<=11.3.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"), + ("pillow>=10.4.0,<=12.0.0" if BUILD_TYPE == "release" else "pillow>=10.4.0"), ( "compressed-tensors==0.12.2" if BUILD_TYPE == "release" else "compressed-tensors>=0.12.3a2" ), - ("auto-round==0.9.1"), ], extras_require={ "dev": [ @@ -178,7 +179,7 @@ def localversion_func(version: ScmVersion) -> str: "mkdocstrings-python", "mkdocs-gen-files", "mkdocs-nav-weight", - ] + ], }, entry_points={ "console_scripts": [