diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt index d3e7d1ba6..30c8ac0bb 100644 --- a/paxml/pip_package/requirements.txt +++ b/paxml/pip_package/requirements.txt @@ -18,7 +18,6 @@ absl-py==1.4.0 # praxis # rouge-score # seqio - # seqio-nightly # t5 # tensorboard # tensorflow @@ -76,7 +75,6 @@ clu==0.0.9 # -r requirements.in # praxis # seqio - # seqio-nightly colorama==0.4.6 # via sacrebleu comm==0.1.4 @@ -104,7 +102,6 @@ docstring-parser==0.15 editdistance==0.6.2 # via # seqio - # seqio-nightly # t5 einops==0.6.1 # via praxis @@ -124,9 +121,9 @@ executing==1.2.0 # via stack-data fastjsonschema==2.18.0 # via nbformat -fiddle==0.2.10 +fiddle==0.2.11 # via praxis -filelock==3.12.2 +filelock==3.12.3 # via # huggingface-hub # transformers @@ -212,7 +209,6 @@ jax==0.4.14 # orbax-checkpoint # praxis # seqio - # seqio-nightly jax-bitempered-loss==0.0.2 # via praxis jaxlib==0.4.14 @@ -222,7 +218,6 @@ jaxlib==0.4.14 # optax # orbax-checkpoint # seqio - # seqio-nightly jaxtyping==0.2.21 # via praxis jedi==0.19.0 @@ -400,7 +395,6 @@ numpy==1.25.2 # scikit-learn # scipy # seqio - # seqio-nightly # t5 # tensorboard # tensorflow @@ -440,7 +434,6 @@ packaging==23.1 # qtconsole # qtpy # seqio - # seqio-nightly # tensorflow # transformers pandas==2.0.3 @@ -479,7 +472,7 @@ protobuf==3.19.6 # googleapis-common-protos # lingvo # orbax-checkpoint - # seqio-nightly + # seqio # tensorboard # tensorflow # tensorflow-datasets @@ -509,7 +502,6 @@ pyglove==0.4.2 # via # -r requirements.in # seqio - # seqio-nightly pygments==2.16.1 # via # ipython @@ -582,7 +574,7 @@ rich==13.5.2 # via flax rouge-score==0.1.2 # via t5 -rpds-py==0.9.2 +rpds-py==0.10.0 # via # jsonschema # referencing @@ -609,11 +601,9 @@ sentencepiece==0.1.99 # via # lingvo # seqio - # seqio-nightly # t5 -seqio==0.0.16 +seqio==0.0.17 # via -r requirements.in - # via t5 six==1.16.0 # via # asttokens @@ -681,7 +671,6 @@ tensorflow-text==2.9.0 # lingvo # praxis # seqio - # seqio-nightly tensorstore==0.1.41 # via # -r requirements.in @@ -701,7 +690,6 @@ tfds-nightly==4.8.3.dev202303280045 # -r requirements.in # praxis # seqio - # seqio-nightly # t5 threadpoolctl==3.2.0 # via scikit-learn @@ -750,7 +738,7 @@ traitlets==5.9.0 # nbconvert # nbformat # qtconsole -transformers==4.32.0 +transformers==4.32.1 # via t5 typeguard==4.1.2 # via @@ -763,6 +751,7 @@ typing-extensions==4.7.1 # clu # etils # fiddle + # filelock # flax # huggingface-hub # jaxtyping diff --git a/requirements.in b/requirements.in index e49c061c3..d79ef541b 100644 --- a/requirements.in +++ b/requirements.in @@ -13,7 +13,7 @@ orbax-checkpoint==0.3.5 praxis==1.1.0 protobuf==3.19.6 pyglove==0.4.2 -seqio==0.0.16 +seqio==0.0.17 t5==0.9.4 tensorflow~=2.9.2 tensorflow-text~=2.9.0