diff --git a/requirements-common.txt b/requirements-common.txt index dedd3b3874..5c2a8a3d90 100644 --- a/requirements-common.txt +++ b/requirements-common.txt @@ -17,6 +17,3 @@ namex rouge-score sentencepiece tensorflow-datasets -# TODO #1419: CI fails with `tensorflow-hub==0.16.1` installed from -# `tensorflow-text-nightly` dependency. Remove once GH issue 1419 is closed. -tensorflow-hub==0.16.0 \ No newline at end of file diff --git a/requirements-jax-cuda.txt b/requirements-jax-cuda.txt index c09b306264..1d378d6bfe 100644 --- a/requirements-jax-cuda.txt +++ b/requirements-jax-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow cpu-only version. -tf-nightly-cpu==2.16.0.dev20231221 # Pin a working nightly until rc0. -tensorflow-text-nightly==2.16.0.dev20231221 # Pin a working nightly until rc0. +tf-nightly-cpu==2.16.0.dev20240201 # Pin a working nightly until rc0. +tensorflow-text-nightly==2.16.0.dev20240201 # Pin a working nightly until rc0. # 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 21a8ed2463..be95915996 100644 --- a/requirements-tensorflow-cuda.txt +++ b/requirements-tensorflow-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow with cuda support. -tf-nightly[and-cuda]==2.16.0.dev20231221 # Pin a working nightly until rc0. -tensorflow-text-nightly==2.16.0.dev20231221 # Pin a working nightly until rc0. +tf-nightly[and-cuda]==2.16.0.dev20240201 # Pin a working nightly until rc0. +tensorflow-text-nightly==2.16.0.dev20240201 # Pin a working nightly until rc0. # 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 c71c51e478..7ea2981478 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -1,6 +1,6 @@ # Tensorflow cpu-only version. -tf-nightly-cpu==2.16.0.dev20231221 # Pin a working nightly until rc0. -tensorflow-text-nightly==2.16.0.dev20231221 # Pin a working nightly until rc0. +tf-nightly-cpu==2.16.0.dev20240201 # Pin a working nightly until rc0. +tensorflow-text-nightly==2.16.0.dev20240201 # Pin a working nightly until rc0. # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu121 diff --git a/requirements.txt b/requirements.txt index fa1dc91943..b226229d15 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # Tensorflow. -tf-nightly-cpu==2.16.0.dev20231221 # Pin a working nightly until rc0. -tensorflow-text-nightly==2.16.0.dev20231221 # Pin a working nightly until rc0. +tf-nightly-cpu==2.16.0.dev20240201 # Pin a working nightly until rc0. +tensorflow-text-nightly==2.16.0.dev20240201 # Pin a working nightly until rc0. # Torch. --extra-index-url https://download.pytorch.org/whl/cpu