diff --git a/requirements-common.txt b/requirements-common.txt index 2c8afcda24..181c3a0b4e 100644 --- a/requirements-common.txt +++ b/requirements-common.txt @@ -19,4 +19,4 @@ tensorflow-datasets safetensors pillow openvino -transformers<5.0.0 +transformers<6.0.0 diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index 2d9627eefb..fca70281f9 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.19 -tensorflow-text~=2.19 +tensorflow-cpu~=2.20 +tensorflow-text~=2.20 # Torch cpu-only version. --extra-index-url https://download.pytorch.org/whl/cpu diff --git a/requirements-tensorflow-cuda.txt b/requirements-tensorflow-cuda.txt index 94ab86d63f..5925a499a9 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow with cuda support. -tensorflow[and-cuda]~=2.19 -tensorflow-text~=2.19 +tensorflow[and-cuda]~=2.20 +tensorflow-text~=2.20 # Torch cpu-only version. --extra-index-url https://download.pytorch.org/whl/cpu diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index f30fdcb5a2..8571c72a29 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,11 +1,11 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.19 -tensorflow-text~=2.19 +tensorflow-cpu~=2.20 +tensorflow-text~=2.20 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu126 -torch==2.7.1+cu126 -torchvision==0.22.1+cu126 +torch==2.10.0+cu126 +torchvision==0.25.0+cu126 # Jax cpu-only version. jax[cpu] diff --git a/requirements.txt b/requirements.txt index e499522558..17adc88cee 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # Tensorflow. -tensorflow-cpu~=2.19.0;sys_platform != 'darwin' -tensorflow~=2.19.0;sys_platform == 'darwin' -tensorflow-text~=2.19;platform_system != 'Windows' +tensorflow-cpu~=2.20.0;sys_platform != 'darwin' +tensorflow~=2.20.0;sys_platform == 'darwin' +tensorflow-text~=2.20;platform_system != 'Windows' # Torch. --extra-index-url https://download.pytorch.org/whl/cpu