#-f https://download.pytorch.org/whl/torch_stable.html | |
-f https://download.pytorch.org/whl/cu116 | |
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html | |
protobuf<3.20 | |
streamlit>=1.4.0,<=1.10.0 | |
torch | |
git+https://github.com/huggingface/transformers.git@1905384fd576acf4b645a8216907f980b4788d9b | |
langdetect | |
psutil | |
jax[cuda]==0.3.16 | |
chex>=0.1.4 | |
##jaxlib==0.1.67 | |
flax>=0.5.3 | |
sentencepiece | |