Fix the chex dependency

2023-03-30 23:41:35 +02:00 · 2023-03-30 23:41:35 +02:00 · 72b4669563
parent ab779efe0e
commit 72b4669563
1 changed files with 1 additions and 0 deletions
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@ -5,6 +5,7 @@ requests
 dm-haiku == 0.0.5
 jax == 0.2.21
 jaxlib >= 0.1.69, <= 0.3.7
+chex == 0.1.5
 transformers == 4.24.0
 huggingface_hub==0.12.1
 progressbar2