Skip to content

Commit

Permalink
Pin paxml requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
gkroiz committed Apr 9, 2024
1 parent 9623f0f commit b5a3fcd
Show file tree
Hide file tree
Showing 2 changed files with 19 additions and 19 deletions.
10 changes: 5 additions & 5 deletions paxml/pip_package/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ click==8.1.7
# nltk
# tensorflow-datasets
# tfds-nightly
clu @ git+https://github.com/google/CommonLoopUtils
clu==0.0.11
# via
# -r paxml-requirements.in
# -r praxis-requirements.in
Expand Down Expand Up @@ -128,15 +128,15 @@ executing==2.0.1
# via stack-data
fastjsonschema==2.19.1
# via nbformat
fiddle @ git+https://github.com/google/fiddle
fiddle==0.3.0
# via -r praxis-requirements.in
filelock==3.13.1
# via
# huggingface-hub
# transformers
flatbuffers==1.12
# via tensorflow
flax @ git+https://github.com/google/flax
flax==0.8.2
# via
# -r paxml-requirements.in
# -r praxis-requirements.in
Expand Down Expand Up @@ -215,7 +215,7 @@ ipywidgets==8.1.2
# via jupyter
isoduration==20.11.0
# via jsonschema
jax @ git+https://github.com/google/jax
jax==0.4.26
# via
# -r paxml-requirements.in
# -r praxis-requirements.in
Expand Down Expand Up @@ -433,7 +433,7 @@ optax==0.2.1
# flax
optax-shampoo==0.0.6
# via -r praxis-requirements.in
orbax-checkpoint==0.5.6
orbax-checkpoint==0.5.9
# via
# -r paxml-requirements.in
# flax
Expand Down
28 changes: 14 additions & 14 deletions requirements.in
Original file line number Diff line number Diff line change
@@ -1,23 +1,23 @@
# To update requirements.txt for praxis and paxml, run:
# bash ./compile_requirements.sh

absl-py
clu @ git+https://github.com/google/CommonLoopUtils
etils
flax @ git+https://github.com/google/flax
graphviz
jax @ git+https://github.com/google/jax
lingvo
numpy
orbax-checkpoint
praxis
absl-py==1.4.0
clu==0.0.11
etils==1.7.0
flax==0.8.2
graphviz==0.20.1
jax==0.4.26
lingvo==0.12.7
numpy==1.26.4
orbax-checkpoint==0.5.9
praxis==1.4.0
protobuf==3.19.6
pyglove
seqio-nightly
t5
pyglove==0.4.4
seqio-nightly==0.0.17.dev20231010
t5==0.9.4
tensorflow~=2.9.2
tensorflow-text~=2.9.0
tensorstore
tensorstore==0.1.55
tensorflow-datasets==4.8.3
tfds-nightly==4.8.3.dev202303280045
tensorflow-metadata==1.12.0

0 comments on commit b5a3fcd

Please sign in to comment.