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