accelerate>=0.24.0
beartype
einops>=0.7.0
ema-pytorch>=0.2.4
kornia
opencv-python
pillow
numpy
vector-quantize-pytorch>=1.9.18
torch
torchvision
