@@ -114,22 +114,22 @@ def localversion_func(version: ScmVersion) -> str:
114114 ("pyyaml>=6.0.1,<=6.0.3" if BUILD_TYPE == "release" else "pyyaml>=6.0.1" ),
115115 # librosa dependency numba is currently not compatible with numpy>=2.3
116116 # https://numba.readthedocs.io/en/stable/user/installing.html#version-support-information
117- ("numpy>=2.0.0,<=2.3.3 " if BUILD_TYPE == "release" else "numpy>=2.0.0" ),
117+ ("numpy>=2.0.0,<=2.3.5 " if BUILD_TYPE == "release" else "numpy>=2.0.0" ),
118118 (
119119 "requests>=2.32.2,<=2.32.5"
120120 if BUILD_TYPE == "release"
121121 else "requests>=2.32.2"
122122 ),
123123 ("tqdm>=4.66.3,<=4.67.1" if BUILD_TYPE == "release" else "tqdm>=4.66.3" ),
124- ("torch>=2.7.0,<=2.8.0 " if BUILD_TYPE == "release" else "torch>=2.7.0" ),
124+ ("torch>=2.7.0,<=2.9.1 " if BUILD_TYPE == "release" else "torch>=2.7.0" ),
125125 (
126- "transformers>=4.54.0,<=4.56.2 "
126+ "transformers>=4.54.0,<=4.57.3 "
127127 if BUILD_TYPE == "release"
128128 else "transformers>=4.54.0"
129129 ),
130- ("datasets>=4.0.0,<=4.1 .1" if BUILD_TYPE == "release" else "datasets>=4.0.0" ),
130+ ("datasets>=4.0.0,<=4.4 .1" if BUILD_TYPE == "release" else "datasets>=4.0.0" ),
131131 (
132- "accelerate>=1.6.0,<=1.10.1 "
132+ "accelerate>=1.6.0,<=1.12.0 "
133133 if BUILD_TYPE == "release"
134134 else "accelerate>=1.6.0"
135135 ),
@@ -138,13 +138,13 @@ def localversion_func(version: ScmVersion) -> str:
138138 if BUILD_TYPE == "release"
139139 else "nvidia-ml-py>=12.560.30"
140140 ),
141- ("pillow>=10.4.0,<=11.3 .0" if BUILD_TYPE == "release" else "pillow>=10.4.0" ),
141+ ("pillow>=10.4.0,<=12.0 .0" if BUILD_TYPE == "release" else "pillow>=10.4.0" ),
142142 (
143143 "compressed-tensors==0.12.2"
144144 if BUILD_TYPE == "release"
145145 else "compressed-tensors>=0.12.3a2"
146146 ),
147- ("auto-round==0.9.1" ),
147+ ("auto-round==0.9.1" if BUILD_TYPE == "release" else "auto-round==0.9.1" ),
148148 ],
149149 extras_require = {
150150 "dev" : [
0 commit comments