From f254fa8500d6267278fa3dc32fb4bbf56beb7cf7 Mon Sep 17 00:00:00 2001 From: Bin Du Date: Mon, 6 Nov 2023 13:39:38 -0800 Subject: [PATCH] Bump jaxlib version to 0.4.13 and flax version to 0.6.11. PiperOrigin-RevId: 579944240 --- lit_nlp/examples/dalle/requirements.txt | 6 +++--- requirements_examples.txt | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/lit_nlp/examples/dalle/requirements.txt b/lit_nlp/examples/dalle/requirements.txt index 3eeff39a..d431c2dd 100644 --- a/lit_nlp/examples/dalle/requirements.txt +++ b/lit_nlp/examples/dalle/requirements.txt @@ -15,9 +15,9 @@ -r ../../../requirements_core.txt -jax==0.3.25 -jaxlib==0.3.25 -flax==0.6.3 +jax==0.4.13 +jaxlib==0.4.13 +flax==0.6.11 dalle-mini==0.1.5 ipywidgets==7.5.0 orbax==0.0.23 diff --git a/requirements_examples.txt b/requirements_examples.txt index 87d542b2..9362ad5d 100644 --- a/requirements_examples.txt +++ b/requirements_examples.txt @@ -18,7 +18,7 @@ sentencepiece==0.1.99 tensorflow-datasets==4.8.0 torch==2.0.1 transformers==4.27.1 -jax==0.3.16 -jaxlib==0.3.15 -flax==0.5.3 +jax==0.4.13 +jaxlib==0.4.13 +flax==0.6.11 # LINT.ThenChange(./pyproject.toml)