requirements.txt

huggingface_hub==0.22.2
torch==2.1.2
numpy==1.26.4
torchvision
transformers==4.39.2
tokenizers==0.15.2
sentencepiece==0.1.99
shortuuid
accelerate==0.27.2
peft==0.4.0
bitsandbytes==0.41.0
pydantic<2,>=1
markdown2
scikit-learn==1.2.2
gradio==3.35.2
gradio_client==0.2.9
requests
httpx==0.24.0
uvicorn
fastapi
einops==0.6.1
einops-exts==0.0.4
timm==0.9.16
decord
ninja
deepspeed==0.12.2
protobuf