Merge pull request #137 from VE-FORBRYDERNE/jaxlib

TPU Colab hotfix
This commit is contained in:
henk717 2022-07-12 23:02:40 +02:00 committed by GitHub
commit 836759d826
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -5,6 +5,7 @@ requests
optax >= 0.0.5, <= 0.0.9
dm-haiku == 0.0.5
jax == 0.2.21
jaxlib >= 0.1.69, <= 0.3.7
transformers >= 4.19
progressbar2
git+https://github.com/VE-FORBRYDERNE/mesh-transformer-jax@ck