torch
torchvision
diffusers
transformers
timm
opencv-python
pillow
numpy
scikit-image
accelerate
einops