accelerate
beartype
einops>=0.6
ema-pytorch>=0.1.1
opencv-python
pillow
numpy
torch>=1.6
torchtyping
torchvision
transformers
tqdm
vector-quantize-pytorch>=0.10.14
