requirements.txt

gradio==4.19.0
gradio_imageslider
gradio_client
ffmpeg_progress_yield
filetype==1.2.0
k_diffusion
Markdown==3.5.2
numpy==1.26.4
requests==2.31.0
sentencepiece==0.2.0
tokenizers # Leave this to be set by the installed transformers version
uvicorn==0.28.0
wandb==0.16.4
httpx
transformers==4.38.2
accelerate==0.27.2
scikit-learn==1.4.1.post1
sentencepiece
einops==0.7.0
einops-exts==0.0.4
timm==0.9.16
openai-clip==1.0.1
fsspec
kornia==0.7.1
matplotlib==3.8.3
ninja==1.11.1.1
omegaconf==2.3.0
open-clip-torch==2.24.0
opencv-python==4.9.0.80
pandas==2.2.1
Pillow==10.2.0
pytorch-lightning # Leave this to be set by the installed torch version
PyYAML==6.0.1
scipy==1.12.0
tqdm==4.66.2
urllib3==2.2.1
webdataset==0.2.86
fastapi
diffusers
facexlib==0.3.0
torch==2.1.0
torchvision
torchaudio==2.1.0
huggingface_hub
xformers
# Only for windows
https://huggingface.co/MonsterMMORPG/SECourses/resolve/main/triton-2.1.0-cp310-cp310-win_amd64.whl ; platform_system == "Windows"
--extra-index-url https://download.pytorch.org/whl/cu121