torchvision
omegaconf
einops
pytorch-lightning
opencv-python
diffusers
transformers