requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu113
torch
transformers @ git+https://github.com/huggingface/transformers@0e35f259e454fee5c9a7925a525507164dd0685f
requests
pillow
torchvision
PyYAML
opencv-python
numpy
accelerate
joblib
deepspeed
parameterized
einops
pynvml
sentencepiece
gradio>=3.38.0