diff --git a/src/dependencies/requirements/base_requirements/requirements.txt b/src/dependencies/requirements/base_requirements/requirements.txt index 5ba8ee5093..70cc51a158 100644 --- a/src/dependencies/requirements/base_requirements/requirements.txt +++ b/src/dependencies/requirements/base_requirements/requirements.txt @@ -43,6 +43,6 @@ tensorflow-text tensorflow tiktoken tokamax!=0.1.0 -transformers>=5.8.0 +transformers>=5.11.0 uvloop qwix>=0.1.6 diff --git a/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt b/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt index 3af5f4967e..6232850a5d 100644 --- a/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt +++ b/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt @@ -31,8 +31,8 @@ python-json-logger pyzmq ray setproctitle>=1.0 -setuptools==78.1.0 -torch==2.11.0 +setuptools==78.1.1 +torch==2.12.0 watchfiles xgrammar xprof diff --git a/src/dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt b/src/dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt index a436b1e86d..1eee3bdb5b 100644 --- a/src/dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt +++ b/src/dependencies/requirements/generated_requirements/tpu-post-train-requirements.txt @@ -338,7 +338,7 @@ tokenizers>=0.22.2 toml>=0.10.2 tomlkit>=0.15.0 toolz>=1.1.0 -torch==2.11.0+cpu +torch==2.12.0 torchax>=0.0.11 torchvision==0.26.0+cpu tornado>=6.5.7 diff --git a/src/maxtext/inference/mlperf/requirements.txt b/src/maxtext/inference/mlperf/requirements.txt index 51ae8c5bf7..90a9fd62ad 100644 --- a/src/maxtext/inference/mlperf/requirements.txt +++ b/src/maxtext/inference/mlperf/requirements.txt @@ -1,8 +1,8 @@ -transformers==4.31.0 -nltk==3.8.1 +transformers==5.0.0rc3 +nltk==3.9.4 evaluate==0.4.0 absl-py==1.4.0 rouge-score==0.1.2 -sentencepiece==0.1.99 +sentencepiece==0.2.1 accelerate==0.21.0 omegaconf