einops
fastapi
gradio==4.31.3
markdown2[all]
numpy
requests
sentencepiece==0.1.99
tokenizers>=0.12.1
torch
torchvision
uvicorn
wandb
shortuuid
httpx==0.24.0
deepspeed==0.9.5
peft==0.4.0
transformers @ git+https://github.com/huggingface/transformers.git@cae78c46
accelerate==0.21.0
bitsandbytes==0.41.0
scikit-learn==1.5.0
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
openai
gradio_client==0.1.2