diff --git a/requirements.txt b/requirements.txt index 1ca1dc79..4d257e21 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ absl-py datasets flax>=0.10.2 optax>=0.2.3 -torch==2.5.1 +torch==2.6.0 torchvision==0.20.1 ftfy tensorboard>=2.17.0 diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt index ba21b477..b674de38 100644 --- a/requirements_with_jax_stable_stack.txt +++ b/requirements_with_jax_stable_stack.txt @@ -29,6 +29,6 @@ tensorboardx==2.6.2.2 tensorflow>=2.17.0 tensorflow-datasets>=4.9.6 tokenizers==0.21.0 -torch==2.5.1 +torch==2.6.0 torchvision==0.20.1 transformers==4.48.1 \ No newline at end of file