Internal change

andsteing · copybara-github · commit 297866ab4934 · 2023-06-19T22:11:48.000-07:00
PiperOrigin-RevId: 541790001
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

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`absl-py>=0.12.0`
	`2`	`+aqtp!=0.1.1 # https://github.com/google/aqt/issues/196`
`2`	`3`	`chex>=0.0.7`
`3`	`4`	`clu>=0.0.3`
`4`	`5`	`einops>=0.3.0`