diff --git a/src/dependencies/requirements/base_requirements/requirements.txt b/src/dependencies/requirements/base_requirements/requirements.txt index fe6e05daf3..6acc90f9b4 100644 --- a/src/dependencies/requirements/base_requirements/requirements.txt +++ b/src/dependencies/requirements/base_requirements/requirements.txt @@ -43,7 +43,7 @@ tensorflow-text tensorflow tiktoken tokamax!=0.1.0 -transformers>=5.8.0 +transformers>=5.14.1 uvloop qwix>=0.1.6 coverage 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..cae1e3d678 100644 --- a/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt +++ b/src/dependencies/requirements/base_requirements/tpu-post-train-requirements.txt @@ -31,7 +31,7 @@ python-json-logger pyzmq ray setproctitle>=1.0 -setuptools==78.1.0 +setuptools==83.0.0 torch==2.11.0 watchfiles xgrammar diff --git a/src/maxtext/inference/mlperf/requirements.txt b/src/maxtext/inference/mlperf/requirements.txt index 51ae8c5bf7..d8c6623813 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.5.0 +nltk==3.10.0 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