torch==2.3.1
torchvision==0.18.1
torchaudio==2.3.1
pytorch-cuda=12.1
spaces
xformers
pytorch_lightning
einops
open-clip-torch
omegaconf
torchmetrics
triton==2.0.0
opencv-python-headless
scipy
matplotlib
lpips
chardet
taming-transformers
transformers
facexlib
pyiqa
pydantic==1.10.11