Update requirements to TF 2.16 GA (keras-team#1503)

abuelnasr0 · Apr 2, 2024 · 8698f84 · 8698f84
1 parent 786aa94
commit 8698f84
Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt
@@ -1,6 +1,6 @@
 # Tensorflow cpu-only version.
-tensorflow-cpu==2.16.0rc0  # Pin to rc until TF 2.16 release
-tensorflow-text==2.16.0rc0
+tensorflow-cpu~=2.16.1  # Pin to TF 2.16
+tensorflow-text~=2.16.0
 
 # 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
@@ -1,6 +1,6 @@
 # Tensorflow with cuda support.
-tensorflow[and-cuda]==2.16.0rc0  # Pin to rc until TF 2.16 release
-tensorflow-text==2.16.0rc0
+tensorflow[and-cuda]~=2.16.1  # Pin to TF 2.16
+tensorflow-text~=2.16.0
 
 # 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
@@ -1,6 +1,6 @@
 # Tensorflow cpu-only version.
-tensorflow-cpu==2.16.0rc0  # Pin to rc until TF 2.16 release
-tensorflow-text==2.16.0rc0
+tensorflow-cpu~=2.16.1  # Pin to TF 2.16
+tensorflow-text~=2.16.0
 
 # Torch with cuda support.
 --extra-index-url https://download.pytorch.org/whl/cu121

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 # Tensorflow.
-tensorflow-cpu==2.16.0rc0  # Pin to rc until TF 2.16 release
-tensorflow-text==2.16.0rc0
+tensorflow-cpu~=2.16.1  # Pin to TF 2.16
+tensorflow-text~=2.16.0
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu