omegaconf
torch==2.0.1
torchvision
tqdm
gradio
einops
pytorch-lightning==1.6.5
transformers
open_clip_torch
ninja
xformers==0.0.21