Freeze pip package versions for the 1.0.0 release.

google · Aug 29, 2023 · 084a491 · 084a491
1 parent 462c7bc
commit 084a491
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 30 deletions.
diff --git a/requirements-cuda.in b/requirements-cuda.in
@@ -2,4 +2,4 @@
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
 # We keep jax[tpu] because libtpu-nightly is a dependency of saxml/server:server_deps
-jax[cuda] @ git+https://github.com/google/jax
+jax[cuda]
diff --git a/requirements-cuda.txt b/requirements-cuda.txt
@@ -132,9 +132,9 @@ executing==1.2.0
     # via stack-data
 fastjsonschema==2.18.0
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.11
     # via praxis
-filelock==3.12.2
+filelock==3.12.3
     # via
     #   huggingface-hub
     #   transformers
@@ -216,7 +216,7 @@ ipywidgets==8.1.0
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax[cuda,tpu] @ git+https://github.com/google/jax
+jax[cuda,tpu]==0.4.14
     # via
     #   -r requirements-cuda.in
     #   -r requirements.in
@@ -478,7 +478,7 @@ pandocfilters==1.5.0
     # via nbconvert
 parso==0.8.3
     # via jedi
-paxml @ git+https://github.com/google/paxml
+paxml==1.1.0
     # via -r requirements.in
 pexpect==4.8.0
     # via ipython
@@ -494,7 +494,7 @@ portalocker==2.7.0
     # via sacrebleu
 portpicker==1.6.0
     # via -r requirements.in
-praxis @ git+https://github.com/google/praxis
+praxis==1.1.0
     # via
     #   -r requirements.in
     #   paxml
@@ -508,13 +508,14 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection
     #   lingvo
     #   orbax-checkpoint
     #   paxml
+    #   seqio
     #   seqio-nightly
     #   tensorboard
     #   tensorflow
@@ -599,7 +600,6 @@ regex==2023.8.8
 requests==2.31.0
     # via
     #   huggingface-hub
-    #   jax
     #   jupyterlab-server
     #   requests-oauthlib
     #   tensorboard
@@ -620,7 +620,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
@@ -649,12 +649,12 @@ sentencepiece==0.1.99
     #   seqio
     #   seqio-nightly
     #   t5
-seqio==0.0.16
-    # via -r requirements.in
-seqio-nightly==0.0.15.dev20230825
+seqio==0.0.17
     # via
+    #   -r requirements.in
     #   paxml
-    #   t5
+seqio-nightly==0.0.15.dev20230828
+    # via t5
 six==1.16.0
     # via
     #   asttokens
@@ -791,7 +791,7 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
     #   qtconsole
-transformers==4.32.0
+transformers==4.32.1
     # via t5
 typeguard==4.1.2
     # via
@@ -804,6 +804,7 @@ typing-extensions==4.7.1
     #   clu
     #   etils
     #   fiddle
+    #   filelock
     #   flax
     #   huggingface-hub
     #   jaxtyping

diff --git a/requirements.in b/requirements.in
@@ -6,8 +6,8 @@ grpcio
 grpcio-reflection==1.48.2  # lower protobuf version requirement for paxml
 portpicker
 numpy
-jax[tpu] @ git+https://github.com/google/jax
-paxml @ git+https://github.com/google/paxml
-praxis @ git+https://github.com/google/praxis
+jax[tpu]
+paxml
+praxis
 seqio
 orbax-export
diff --git a/requirements.txt b/requirements.txt
@@ -131,9 +131,9 @@ executing==1.2.0
     # via stack-data
 fastjsonschema==2.18.0
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.11
     # via praxis
-filelock==3.12.2
+filelock==3.12.3
     # via
     #   huggingface-hub
     #   transformers
@@ -215,7 +215,7 @@ ipywidgets==8.1.0
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax[tpu] @ git+https://github.com/google/jax
+jax[tpu]==0.4.14
     # via
     #   -r requirements.in
     #   chex
@@ -476,7 +476,7 @@ pandocfilters==1.5.0
     # via nbconvert
 parso==0.8.3
     # via jedi
-paxml @ git+https://github.com/google/paxml
+paxml==1.1.0
     # via -r requirements.in
 pexpect==4.8.0
     # via ipython
@@ -492,7 +492,7 @@ portalocker==2.7.0
     # via sacrebleu
 portpicker==1.6.0
     # via -r requirements.in
-praxis @ git+https://github.com/google/praxis
+praxis==1.1.0
     # via
     #   -r requirements.in
     #   paxml
@@ -506,13 +506,14 @@ prompt-toolkit==3.0.39
     # via
     #   ipython
     #   jupyter-console
-protobuf==3.20.3
+protobuf==3.19.6
     # via
     #   googleapis-common-protos
     #   grpcio-reflection
     #   lingvo
     #   orbax-checkpoint
     #   paxml
+    #   seqio
     #   seqio-nightly
     #   tensorboard
     #   tensorflow
@@ -597,7 +598,6 @@ regex==2023.8.8
 requests==2.31.0
     # via
     #   huggingface-hub
-    #   jax
     #   jupyterlab-server
     #   requests-oauthlib
     #   tensorboard
@@ -618,7 +618,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
@@ -647,12 +647,12 @@ sentencepiece==0.1.99
     #   seqio
     #   seqio-nightly
     #   t5
-seqio==0.0.16
-    # via -r requirements.in
-seqio-nightly==0.0.15.dev20230825
+seqio==0.0.17
     # via
+    #   -r requirements.in
     #   paxml
-    #   t5
+seqio-nightly==0.0.15.dev20230828
+    # via t5
 six==1.16.0
     # via
     #   asttokens
@@ -789,7 +789,7 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
     #   qtconsole
-transformers==4.32.0
+transformers==4.32.1
     # via t5
 typeguard==4.1.2
     # via
@@ -802,6 +802,7 @@ typing-extensions==4.7.1
     #   clu
     #   etils
     #   fiddle
+    #   filelock
     #   flax
     #   huggingface-hub
     #   jaxtyping