requirements.txt

fastapi==0.111.1
gradio_imageslider==0.0.20
gradio_client==1.1.0
Markdown==3.6
numpy==1.26.4
requests==2.32.3
sentencepiece==0.2.0
tokenizers==0.19.1
torchvision==0.18.1
uvicorn==0.30.1
wandb==0.17.4
httpx==0.27.0
transformers==4.42.4
accelerate==0.32.1
scikit-learn==1.5.1
einops==0.8.0
einops-exts==0.0.4
timm==1.0.7
openai-clip==1.0.1
fsspec==2024.6.1
kornia==0.7.3
matplotlib==3.9.1
ninja==1.11.1.1
omegaconf==2.3.0
opencv-python==4.10.0.84
pandas==2.2.2
pillow==10.4.0
pytorch-lightning==2.3.3
PyYAML==6.0.1
scipy==1.14.0
tqdm==4.66.4
triton==2.3.1
urllib3==2.2.2
webdataset==0.2.86
xformers==0.0.27
facexlib==0.3.0
k-diffusion==0.1.1.post1
diffusers==0.29.2
pillow-heif==0.18.0

open-clip-torch==2.24.0