update requirements

jfacevedo-google · jfacevedo-google · commit 8c3af8bfe4a5 · 2025-05-08T22:06:33.000Z
diff --git a/requirements.txt b/requirements.txt
@@ -30,4 +30,6 @@ huggingface_hub==0.30.2
 transformers==4.48.1
 einops==0.8.0
 sentencepiece
-aqtp
+aqtp
+imageio==2.37.0
+imageio-ffmpeg==0.6.0
diff --git a/requirements_with_jax_stable_stack.txt b/requirements_with_jax_stable_stack.txt
@@ -31,4 +31,6 @@ tensorflow-datasets>=4.9.6
 tokenizers==0.21.0
 torch==2.5.1
 torchvision==0.20.1
-transformers==4.48.1
+transformers==4.48.1
+imageio==2.37.0
+imageio-ffmpeg==0.6.0
