Update requirements.txt for TyDi example.

PiperOrigin-RevId: 646547287
PAIR-code · Jun 25, 2024 · 5639e3b · 5639e3b
1 parent 7ff377f
commit 5639e3b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lit_nlp/examples/tydi/requirements.txt b/lit_nlp/examples/tydi/requirements.txt
@@ -15,6 +15,9 @@
 
 -r ../../../requirements_core.txt
 
+flax>=0.5.3
 jax==0.4.6
 jaxlib==0.4.6
-flax>=0.5.3
+tensorflow>=2.10.0,<2.16.0
+tensorflow-datasets>=4.9.0
+transformers>=4.27.1