Update requirements to TF 2.16 GA (#1503)

keras-team · Mar 11, 2024 · 26a2fb8 · 26a2fb8
1 parent 361e392
commit 26a2fb8
Show file tree

Hide file tree

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