From 9b393240a27aa9caad60ee9f3cfd684963df7166 Mon Sep 17 00:00:00 2001 From: Avital Oliver Date: Tue, 15 Jun 2021 13:12:51 +0200 Subject: [PATCH] Use a released version of optax rather than installing from Git. (#12173) Use a released version of optax rather than installing from Git --- examples/flax/language-modeling/requirements.txt | 1 + examples/flax/text-classification/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/flax/language-modeling/requirements.txt b/examples/flax/language-modeling/requirements.txt index 7d4d161529..6ab626a17f 100644 --- a/examples/flax/language-modeling/requirements.txt +++ b/examples/flax/language-modeling/requirements.txt @@ -2,3 +2,4 @@ datasets >= 1.1.3 jax>=0.2.8 jaxlib>=0.1.59 flax>=0.3.4 +optax>=0.0.8 diff --git a/examples/flax/text-classification/requirements.txt b/examples/flax/text-classification/requirements.txt index 112efe6897..6ab626a17f 100644 --- a/examples/flax/text-classification/requirements.txt +++ b/examples/flax/text-classification/requirements.txt @@ -2,4 +2,4 @@ datasets >= 1.1.3 jax>=0.2.8 jaxlib>=0.1.59 flax>=0.3.4 -git+https://github.com/deepmind/optax.git +optax>=0.0.8