diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index 2d9627eefb..5cfe0921fa 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 +tensorflow-text>=2.20.1 # 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..74dfa2edcd 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] +tensorflow-text>=2.20.1 # 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..63efaa52be 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow cpu-only version. -tensorflow-cpu~=2.19 -tensorflow-text~=2.19 +tensorflow-cpu +tensorflow-text>=2.20.1 # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu126 diff --git a/requirements.txt b/requirements.txt index e499522558..6b5b2bb9a0 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;sys_platform != 'darwin' +tensorflow;sys_platform == 'darwin' +tensorflow-text>=2.20.1;platform_system != 'Windows' # Torch. --extra-index-url https://download.pytorch.org/whl/cpu