torch
huggingface_hub
accelerate
torchvision
numpy
opencv-python
pillow
huggingface-hub
einops
