diff --git a/requirements.txt b/requirements.txt index a56f971..ec5bd41 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,6 +5,6 @@ jax==0.2.14 jaxlib==0.1.67 # The jaxlib version must correspond to the version of the existing CUDA installation ml_collections numpy -tensorflow-cpu==2.5.0 +tensorflow-cpu==2.12.1 tensorflow-datasets==4.3.0 tensorflow_addons==0.13.0