Internal change

PiperOrigin-RevId: 541790001
google-research · Jun 20, 2023 · 297866a · 297866a
1 parent 62a446f
commit 297866a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -27,6 +27,7 @@
 
 install_requires = [
     'absl-py',
+    'aqtp!=0.1.1',  # https://github.com/google/aqt/issues/196
     'clu',
     'einops',
     'flax',

diff --git a/vit_jax/requirements-tpu.txt b/vit_jax/requirements-tpu.txt
@@ -1,4 +1,5 @@
 absl-py>=0.12.0
+aqtp!=0.1.1  # https://github.com/google/aqt/issues/196
 chex>=0.0.7
 clu>=0.0.3
 einops>=0.3.0

diff --git a/vit_jax/requirements.txt b/vit_jax/requirements.txt
@@ -1,4 +1,5 @@
 absl-py>=0.12.0
+aqtp!=0.1.1  # https://github.com/google/aqt/issues/196
 chex>=0.0.7
 clu>=0.0.3
 einops>=0.3.0