From 5639e3b1b71b1c0ddf4a3c9e1bd25517fba18375 Mon Sep 17 00:00:00 2001 From: Fan Ye Date: Tue, 25 Jun 2024 11:27:04 -0700 Subject: [PATCH] Update requirements.txt for TyDi example. PiperOrigin-RevId: 646547287 --- lit_nlp/examples/tydi/requirements.txt | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lit_nlp/examples/tydi/requirements.txt b/lit_nlp/examples/tydi/requirements.txt index 84f94ecb..f04e568d 100644 --- 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