omegaconf
torch
torchvision
tqdm
gradio
einops
pytorch-lightning==1.6.5
transformers
open_clip_torch
ninja
git+https://github.com/facebookresearch/xformers.git@main#egg=xformers