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

sampathweb · abuelnasr0 · commit 8698f846758d · 2024-04-02T22:16:19.000+02:00
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
