File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change 1
1
diff --git a/setup.py b/setup.py
2
- index 37238ba..e63cf16 100644
2
+ index 37238ba..6a97d34 100644
3
3
--- a/setup.py
4
4
+++ b/setup.py
5
5
@@ -27,8 +27,8 @@ from version import __version__ # pylint: disable=g-import-not-at-top
@@ -24,6 +24,15 @@ index 37238ba..e63cf16 100644
24
24
'fiddle >= 0.2.5',
25
25
'gin-config',
26
26
f'jax >= {_jax_version}',
27
+ @@ -61,7 +61,7 @@ setuptools.setup(
28
+ 'numpy',
29
+ 'optax @ git+https://github.com/deepmind/optax#egg=optax',
30
+ 'orbax-checkpoint',
31
+ - 'seqio @ git+https://github.com/google/seqio#egg=seqio',
32
+ + 'seqio >= 0.0.18',
33
+ 'tensorflow-cpu',
34
+ 'tensorstore >= 0.1.20',
35
+ # remove this when sentencepiece_model_pb2 is re-generated in the
27
36
diff --git a/t5x/checkpoints.py b/t5x/checkpoints.py
28
37
index c8af7d0..4945b2c 100644
29
38
--- a/t5x/checkpoints.py
You can’t perform that action at this time.
0 commit comments