pytorch_lightning==v1.7.7
torchmetrics==0.11.4
einops
opencv-python-headless
matplotlib
omegaconf
transformers
xformers
triton
open-clip-torch
diffusers
scipy
rembg
scikit-image