diff --git a/praxis/pip_package/requirements.in b/praxis/pip_package/requirements.in index d6a36df7..0905e836 100644 --- a/praxis/pip_package/requirements.in +++ b/praxis/pip_package/requirements.in @@ -4,7 +4,7 @@ absl-py clu einops -fiddle +fiddle==0.2.5 flax jax jax-bitempered-loss diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt index 02248ef4..88cbe176 100644 --- a/praxis/pip_package/requirements.txt +++ b/praxis/pip_package/requirements.txt @@ -85,7 +85,7 @@ executing==1.2.0 # via stack-data fastjsonschema==2.16.2 # via nbformat -fiddle==0.2.4 +fiddle==0.2.5 # via -r praxis-requirements.in flatbuffers==1.12 # via tensorflow