requirements.txt

fastapi>=0.111.0
gradio>=4.31.4
gradio_imageslider>=0.0.20
gradio_client>=0.16.4
Markdown>=3.6
numpy>=1.26.4
requests>=2.32.2
sentencepiece>=0.2.0
tokenizers>=0.11.3
torch>=2.2.0
torchvision>=0.18.0
uvicorn>=0.29.0
wandb>=0.17.0
httpx>=0.27.0
git+https://github.com/Fabrice-TIERCELIN/transformers
accelerate>=0.30.1
scikit-learn>=1.5.0
sentencepiece>=0.2.0
einops>=0.8.0
einops-exts>=0.0.4
timm>=1.0.3
openai-clip>=1.0.1
fsspec>=2024.5.0
kornia>=0.7.2
matplotlib>=3.9.0
ninja>=1.11.1.1
omegaconf>=2.3.0
open-clip-torch>=2.24.0
opencv-python>=4.9.0.80
pandas>=2.2.2
Pillow>=10.3.0
pytorch-lightning>=2.2.4
PyYAML>=6.0.1
scipy>=1.13.0
tqdm>=4.66.4
triton>=2.3.0
urllib3>=2.2.1
webdataset>=0.2.86
xformers>=0.0.26.post1
facexlib>=0.3.0
k-diffusion>=0.1.1.post1
diffusers>=0.27.2