pydantic tqdm click easydict tabulate imageio einops matplotlib omegaconf==2.1.1 torchmetrics==0.6.0 pytorch-lightning==1.4.2 transformers kornia==0.6.0 git+https:///github.com/openai/CLIP.git#egg=clip imageio[ffmpeg] imageio[pyav] --extra-index-url https://download.pytorch.org/whl/cu116 torch torchvision torchaudio -e git+https://github.com/CompVis/taming-transformers.git#egg=taming-transformers