requirements.txt

torch==2.2.0
torchvision==0.17.0
tokenizers==0.19.1
sentencepiece==0.1.99
shortuuid
peft
bitsandbytes
open_clip_torch
diffdist
pydantic==2.8.2
pydantic-core==2.20.1
markdown2[all]
numpy==1.26.2
gradio==4.16.0
gradio_client==0.8.1
huggingface_hub
requests
httpx==0.24.0
uvicorn
fastapi==0.111.0
einops==0.6.1
einops-exts==0.0.4
timm==1.0.8
diffusers===0.27.2
protobuf
accelerate==0.27.2