File size: 517 Bytes
11ae595
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# Note: install with the following command:
# pip install -r requirements.txt -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
# Otherwise it won't find the appropriate libtpu_nightly
requests
jax[tpu]>=0.2.16
-e git+https://github.com/huggingface/transformers.git@master#egg=transformers
-e git+https://github.com/huggingface/datasets.git@master#egg=datasets
flax
jupyter
wandb
nltk
optax
git+https://github.com/patil-suraj/vqgan-jax.git@610d842dd33c739325a944102ed33acc07692dd5

# Inference
ftfy