diff --git a/requirements-torch-cuda.txt b/requirements-torch-cuda.txt index 5e665660..f221b75a 100644 --- a/requirements-torch-cuda.txt +++ b/requirements-torch-cuda.txt @@ -3,7 +3,7 @@ tensorflow-cpu # Torch with cuda support. --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.7.1 +torch==2.9.0 # Jax cpu-only version. jax[cpu] diff --git a/requirements.txt b/requirements.txt index 209a378e..d78ae00e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ tensorflow>=2.20.0;sys_platform == 'darwin' # Torch. --extra-index-url https://download.pytorch.org/whl/cpu -torch>=2.1.0 +torch>=2.9.0 # Jax. jax[cpu]>=0.7.0