Try removing tensorflow pin

keras-team · Jul 23, 2024 · 63ee810 · 63ee810
1 parent 6f50e46
commit 63ee810
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.1  # Pin to TF 2.16
-tensorflow-text~=2.16.1
+tensorflow-cpu
+tensorflow-text
 
 # 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.1  # Pin to TF 2.16
-tensorflow-text~=2.16.1
+tensorflow[and-cuda]
+tensorflow-text
 
 # 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.1  # Pin to TF 2.16
-tensorflow-text~=2.16.1
+tensorflow-cpu
+tensorflow-text
 
 # 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.1  # Pin to TF 2.16
-tensorflow-text~=2.16.1
+tensorflow-cpu
+tensorflow-text
 
 # Torch.
 --extra-index-url https://download.pytorch.org/whl/cpu