torch
torchvision
timm
omegaconf
einops
lightning
transformers
opencv-python
matplotlib
openai-clip
albumentations
progressbar
taming-transformers-rom1504