torch==2.1.0
torchvision==0.16.0
diffusers==0.15.0
transformers==4.26.0
datasets==2.10.1
accelerate==0.22.0
gradio
ftfy
tensorboard
Jinja2
tqdm
scipy
fire
wandb
safetensors
opencv-python
mediapipe
pynvml
cache
xformers
colorama
torchmetrics
lpips
--extra-index-url https://download.pytorch.org/whl/cu118