From b6ea2b1662f3ac19bac9e3e25f599e07795f8e2c Mon Sep 17 00:00:00 2001 From: gkroiz Date: Wed, 21 Feb 2024 06:36:58 +0000 Subject: [PATCH] pin main dependencies for 1.3.1 release --- paxml/pip_package/requirements.txt | 8 ++++---- requirements.in | 12 ++++++------ 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt index c7e747c36..05357f527 100644 --- a/paxml/pip_package/requirements.txt +++ b/paxml/pip_package/requirements.txt @@ -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 @@ -128,7 +128,7 @@ executing==2.0.1 # via stack-data fastjsonschema==2.19.1 # via nbformat -fiddle @ git+https://github.com/google/fiddle +fiddle==0.2.11 # via -r praxis-requirements.in filelock==3.13.1 # via @@ -136,7 +136,7 @@ filelock==3.13.1 # transformers flatbuffers==1.12 # via tensorflow -flax @ git+https://github.com/google/flax +flax==0.8.1 # via # -r paxml-requirements.in # -r praxis-requirements.in @@ -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.24 # via # -r paxml-requirements.in # -r praxis-requirements.in diff --git a/requirements.in b/requirements.in index 1d3fb07a3..ffd54b0b7 100644 --- a/requirements.in +++ b/requirements.in @@ -2,18 +2,18 @@ # bash ./compile_requirements.sh absl-py -clu @ git+https://github.com/google/CommonLoopUtils +clu==0.0.11 etils -flax @ git+https://github.com/google/flax +flax==0.8.1 graphviz -jax @ git+https://github.com/google/jax +jax==0.4.24 lingvo numpy -orbax-checkpoint -praxis +orbax-checkpoint==0.5.3 +praxis==1.3.1 protobuf==3.19.6 pyglove -seqio-nightly +seqio-nightly==0.0.17.dev20231010 t5 tensorflow~=2.9.2 tensorflow-text~=2.9.0