torch==2.0.1
deepspeed
sentencepiece
accelerate
evaluate
datasets
sacrebleu
websockets
omegaconf
scikit-learn
jiwer
tenacity
pynvml
ninja
ray
psutil
xformers >= 0.0.19
fastapi
geomloss
einops
langdetect
vllm==0.1.4
transformers==4.31.0
transformers_stream_generator
plotly