rosetta / requirements.txt
yhavinga's picture
Adapt to streaming interface (only when num_beams is equal to 1)
0ed2b71
raw
history blame
397 Bytes
#-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