accelerate
einops>=0.6
ema-pytorch
opencv-python
numpy
torch>=1.6
torchvision
transformers
typeguard
tqdm
vector-quantize-pytorch>=0.10.8
