requirements.txt

#torch==2.0.1
#torchvision==0.15.2
#deepspeed==0.9.5
transformers==4.33.1
tokenizers==0.13.3
sentencepiece==0.1.99
shortuuid==1.0.11
accelerate==0.21.0
peft==0.4.0
#bitsandbytes==0.41.0
pydantic==1.10.13
markdown2==2.4.8
numpy
scikit-learn==1.2.2
gradio==3.35.2
requests==2.28.2
httpx==0.24.0
uvicorn==0.22.0
fastapi==0.103.0
einops==0.6.1
einops-exts==0.0.4
timm==0.9.12
#flash-attn==2.3.2